lattice/pyerrors
1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-05-14 19:43:41 +02:00
Code Issues Projects Releases Packages Wiki Activity

tests: tests for reconstruction of prange from json format added

Browse source
This commit is contained in:
Fabian Joswig 2022-01-28 12:41:55 +00:00
parent 6cdb9a2042
commit 1d41a664ee
1 changed files with 17 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/io_test.py
View file

@ -98,11 +98,11 @@ def test_json_corr_io():
for obs_list in [my_list, rw_list]: for obs_list in [my_list, rw_list]:
for tag in [None, "test"]: for tag in [None, "test"]:
obs_list[3].tag = tag obs_list[3].tag = tag
for fp in [0, 2]: for pad in [0, 2]:
for bp in [0, 7]:
for corr_tag in [None, 'my_Corr_tag']: for corr_tag in [None, 'my_Corr_tag']:
for prange in [None, [3, 6]]:
for gap in [False, True]: for gap in [False, True]:
my_corr = pe.Corr(obs_list, padding=[fp, bp]) my_corr = pe.Corr(obs_list, padding=[pad, pad], prange=prange)
my_corr.tag = corr_tag my_corr.tag = corr_tag
if gap: if gap:
my_corr.content[4] = None my_corr.content[4] = None
@ -114,6 +114,7 @@ def test_json_corr_io():
if entry is None: if entry is None:
assert recover[index] is None assert recover[index] is None
assert my_corr.tag == recover.tag assert my_corr.tag == recover.tag
assert my_corr.prange == recover.prange
assert my_corr.reweighted == recover.reweighted assert my_corr.reweighted == recover.reweighted
@ -123,7 +124,8 @@ def test_json_corr_2d_io():
for tag in [None, "test"]: for tag in [None, "test"]:
obs_list[3][0, 1].tag = tag obs_list[3][0, 1].tag = tag
for padding in [0, 1]: for padding in [0, 1]:
my_corr = pe.Corr(obs_list, padding=[padding, padding]) for prange in [None, [3, 6]]:
my_corr = pe.Corr(obs_list, padding=[padding, padding], prange=prange)
my_corr.tag = tag my_corr.tag = tag
pe.input.json.dump_to_json(my_corr, 'corr') pe.input.json.dump_to_json(my_corr, 'corr')
recover = pe.input.json.load_json('corr') recover = pe.input.json.load_json('corr')
@ -133,3 +135,4 @@ def test_json_corr_2d_io():
if entry is None: if entry is None:
assert recover[index] is None assert recover[index] is None
assert my_corr.tag == recover.tag assert my_corr.tag == recover.tag
assert my_corr.prange == recover.prange