close but need to convert yield tests

openvax · Jan 23, 2024 · ac5e9d1 · ac5e9d1
1 parent 581fcef
commit ac5e9d1
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -76,5 +76,8 @@ def expect_effect(
             format_string = "Expected %s='%s' but got '%s'"
         assert actual_value == expected_value, format_string % (field, expected_value, actual_value)
 
-def eq_(x, y):
-    assert x == y, "%s != %s" % (x, y)
+def eq_(x, y, s=None):
+    if s is None:
+        assert x == y, "%s != %s" % (x, y)
+    else:
+        assert x == y, s