From 01c3d02394e90d83a8b68764bf90fb792eaf1b9d Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 27 Jan 2022 14:07:27 +0000 Subject: [PATCH] tests: tests of json corr format extended --- tests/io_test.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tests/io_test.py b/tests/io_test.py index 6b5c0917..f069e177 100644 --- a/tests/io_test.py +++ b/tests/io_test.py @@ -107,6 +107,9 @@ def test_json_corr_io(): 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]]) + for index, entry in enumerate(my_corr): + if entry is None: + assert recover[index] is None assert my_corr.tag == recover.tag assert my_corr.reweighted == recover.reweighted @@ -123,4 +126,7 @@ def test_json_corr_2d_io(): 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]]) + for index, entry in enumerate(my_corr): + if entry is None: + assert recover[index] is None assert my_corr.tag == recover.tag