From 6c99e05e5bf7df38e9b0828446091647e093b998 Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Thu, 27 Jan 2022 14:12:10 +0000 Subject: [PATCH] test: test of json output of a correlator with a gap added --- tests/io_test.py | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/tests/io_test.py b/tests/io_test.py index f069e177..b6989f60 100644 --- a/tests/io_test.py +++ b/tests/io_test.py @@ -101,17 +101,20 @@ def test_json_corr_io(): for fp in [0, 2]: for bp in [0, 7]: for corr_tag in [None, 'my_Corr_tag']: - my_corr = pe.Corr(obs_list, padding=[fp, bp]) - 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]]) - 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 + for gap in [False, True]: + my_corr = pe.Corr(obs_list, padding=[fp, bp]) + my_corr.tag = corr_tag + if gap: + my_corr.content[4] = None + 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]]) + 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 def test_json_corr_2d_io():