mirror of https://github.com/sgoudham/Enso-Bot.git
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.
64 lines
1.2 KiB
Python
64 lines
1.2 KiB
Python
5 years ago
|
#!/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()
|