test: be more careful about creating paths

kalekundert · Jan 31, 2022 · 6e6dbc3 · 6e6dbc3
1 parent a6fe696
commit 6e6dbc3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_plot.py b/tests/test_plot.py
@@ -66,8 +66,9 @@ def test_pick_attrs(df, attrs, expected, error):
         indirect=['layout'],
 )
 def test_show(layout, attrs, color, tol, expected, error):
-    actual = layout.parent / expected
-    expected = REF_IMAGES / expected
+    if not error:
+        actual = layout.parent / expected
+        expected = REF_IMAGES / expected
 
     with error:
         fig = show(layout, attrs, color)