From c8ec5909f14e760bf6bab17140eb52729eafb721 Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Wed, 19 Jan 2022 10:46:33 +0000 Subject: [PATCH] fix: io tests for corr now properly clean up --- tests/io_test.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/io_test.py b/tests/io_test.py index b9726130..6b5c0917 100644 --- a/tests/io_test.py +++ b/tests/io_test.py @@ -105,6 +105,7 @@ def test_json_corr_io(): my_corr.tag = corr_tag pe.input.json.dump_to_json(my_corr, 'corr') recover = pe.input.json.load_json('corr') + os.remove('corr.json.gz') assert np.all([o.is_zero() for o in [x for x in (my_corr - recover) if x is not None]]) assert my_corr.tag == recover.tag assert my_corr.reweighted == recover.reweighted @@ -120,5 +121,6 @@ def test_json_corr_2d_io(): my_corr.tag = tag pe.input.json.dump_to_json(my_corr, 'corr') recover = pe.input.json.load_json('corr') + os.remove('corr.json.gz') assert np.all([np.all([o.is_zero() for o in q]) for q in [x.ravel() for x in (my_corr - recover) if x is not None]]) assert my_corr.tag == recover.tag