fix

scikit-hep · Oct 10, 2024 · 797d4e6 · 797d4e6
1 parent f3dddb5
commit 797d4e6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/pyhepmc/view.py b/src/pyhepmc/view.py
@@ -255,6 +255,7 @@ def savefig(
         except ValueError:
             if p.exists():
                 p.unlink()
+            raise
         return
 
     # if we arrive here, fname is a file-like object

diff --git a/tests/test_view.py b/tests/test_view.py
@@ -121,13 +121,17 @@ def test_savefig_1(evt, ext):
             assert np.sum(a1[:size] != a2[:size]) < 150
 
 
-def test_savefig_2(evt):
+def test_savefig_2a(evt):
     with pytest.raises(ValueError):
         view.savefig(evt, "foo")
 
+
+def test_savefig_2b(evt):
     with pytest.raises(ValueError):
         view.savefig(evt, "foo.foo")
 
+
+def test_savefig_2c(evt):
     with pytest.raises(ValueError):
         with io.BytesIO() as f:
             view.savefig(evt, f)