You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Enso-Bot/venv/Lib/site-packages/dill/tests/test_check.py

64 lines
1.2 KiB
Python

#!/usr/bin/env python
#
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
# Copyright (c) 2008-2016 California Institute of Technology.
# Copyright (c) 2016-2019 The Uncertainty Quantification Foundation.
# License: 3-clause BSD. The full license text is available at:
# - https://github.com/uqfoundation/dill/blob/master/LICENSE
from dill import check
import sys
from dill.temp import capture
from dill._dill import PY3
#FIXME: this doesn't catch output... it's from the internal call
def raise_check(func, **kwds):
try:
with capture('stdout') as out:
check(func, **kwds)
except Exception:
e = sys.exc_info()[1]
raise AssertionError(str(e))
else:
assert 'Traceback' not in out.getvalue()
finally:
out.close()
f = lambda x:x**2
def test_simple():
raise_check(f)
def test_recurse():
raise_check(f, recurse=True)
def test_byref():
raise_check(f, byref=True)
def test_protocol():
raise_check(f, protocol=True)
def test_python():
raise_check(f, python=None)
#TODO: test incompatible versions
#TODO: test dump failure
#TODO: test load failure
if __name__ == '__main__':
test_simple()
test_recurse()
test_byref()
test_protocol()
test_python()