fix (some) failing tests

lint
pschanely · Dec 19, 2024 · 6ebde1e · 6ebde1e
1 parent 494a2c2
commit 6ebde1e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/crosshair/diff_behavior.py b/crosshair/diff_behavior.py
@@ -295,12 +295,12 @@ def run_iteration(
             realized_args,
             Result(
                 repr(deep_realize(result1[0])),
-                realize(repr(result1[1])),
+                realize(repr(result1[1]) if result1[1] else None),
                 post_execution_args1,
             ),
             Result(
                 repr(deep_realize(result2[0])),
-                realize(repr(result2[1])),
+                realize(repr(result2[1]) if result2[1] else None),
                 post_execution_args2,
             ),
             coverage_manager.get_results(fn1),

diff --git a/crosshair/main_test.py b/crosshair/main_test.py
@@ -61,7 +61,12 @@ def call_check(
 def call_diffbehavior(fn1: str, fn2: str) -> Tuple[int, List[str]]:
     buf: io.StringIO = io.StringIO()
     errbuf: io.StringIO = io.StringIO()
-    retcode = diffbehavior(Namespace(fn1=fn1, fn2=fn2), DEFAULT_OPTIONS, buf, errbuf)
+    retcode = diffbehavior(
+        Namespace(fn1=fn1, fn2=fn2, exception_equivalence="type_and_message"),
+        DEFAULT_OPTIONS,
+        buf,
+        errbuf,
+    )
     lines = [
         ls for ls in buf.getvalue().split("\n") + errbuf.getvalue().split("\n") if ls
     ]