diff options
Diffstat (limited to '')
-rw-r--r-- | src/test/pybind/assertions.py | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/test/pybind/assertions.py b/src/test/pybind/assertions.py new file mode 100644 index 000000000..719700f3a --- /dev/null +++ b/src/test/pybind/assertions.py @@ -0,0 +1,26 @@ +def assert_equal(a, b): + assert a == b + +def assert_not_equal(a, b): + assert a != b + +def assert_greater(a, b): + assert a > b + +def assert_greater_equal(a, b): + assert a >= b + +def assert_raises(excClass, callableObj, *args, **kwargs): + """ + Like unittest.TestCase.assertRaises, but returns the exception. + """ + try: + callableObj(*args, **kwargs) + except excClass as e: + return e + else: + if hasattr(excClass, '__name__'): + excName = excClass.__name__ + else: + excName = str(excClass) + raise AssertionError("%s not raised" % excName) |