summaryrefslogtreecommitdiffstats
path: root/src/test/pybind/assertions.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/pybind/assertions.py')
-rw-r--r--src/test/pybind/assertions.py26
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)