hypothesishelpers.py
71 lines
| 2.1 KiB
| text/x-python
|
PythonLexer
/ tests / hypothesishelpers.py
|
r26842 | # Helper module to use the Hypothesis tool in tests | ||
# | ||||
# Copyright 2015 David R. MacIver | ||||
# | ||||
# For details see http://hypothesis.readthedocs.org | ||||
|
r28728 | from __future__ import absolute_import, print_function | ||
|
r26842 | import os | ||
import sys | ||||
import traceback | ||||
|
r27998 | try: | ||
# hypothesis 2.x | ||||
from hypothesis.configuration import set_hypothesis_home_dir | ||||
from hypothesis import settings | ||||
except ImportError: | ||||
# hypothesis 1.x | ||||
from hypothesis.settings import set_hypothesis_home_dir | ||||
from hypothesis import Settings as settings | ||||
|
r26842 | import hypothesis.strategies as st | ||
|
r27998 | from hypothesis import given | ||
|
r26842 | |||
# hypothesis store data regarding generate example and code | ||||
set_hypothesis_home_dir(os.path.join( | ||||
os.getenv('TESTTMP'), ".hypothesis" | ||||
)) | ||||
def check(*args, **kwargs): | ||||
"""decorator to make a function a hypothesis test | ||||
Decorated function are run immediately (to be used doctest style)""" | ||||
def accept(f): | ||||
# Workaround for https://github.com/DRMacIver/hypothesis/issues/206 | ||||
# Fixed in version 1.13 (released 2015 october 29th) | ||||
f.__module__ = '__anon__' | ||||
try: | ||||
|
r27999 | with settings(max_examples=2000): | ||
given(*args, **kwargs)(f)() | ||||
|
r26842 | except Exception: | ||
traceback.print_exc(file=sys.stdout) | ||||
sys.exit(1) | ||||
return accept | ||||
def roundtrips(data, decode, encode): | ||||
"""helper to tests function that must do proper encode/decode roundtripping | ||||
""" | ||||
@given(data) | ||||
def testroundtrips(value): | ||||
encoded = encode(value) | ||||
decoded = decode(encoded) | ||||
if decoded != value: | ||||
raise ValueError( | ||||
"Round trip failed: %s(%r) -> %s(%r) -> %r" % ( | ||||
encode.__name__, value, decode.__name__, encoded, | ||||
decoded | ||||
)) | ||||
try: | ||||
testroundtrips() | ||||
except Exception: | ||||
# heredoc swallow traceback, we work around it | ||||
traceback.print_exc(file=sys.stdout) | ||||
raise | ||||
print("Round trip OK") | ||||
# strategy for generating bytestring that might be an issue for Mercurial | ||||
bytestrings = ( | ||||
st.builds(lambda s, e: s.encode(e), st.text(), st.sampled_from([ | ||||
'utf-8', 'utf-16', | ||||
]))) | st.binary() | ||||