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

test: json io test extended

Browse source
This commit is contained in:
Fabian Joswig 2021-12-10 10:43:52 +00:00
parent f6dc78f587
commit f7e64b2d38
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/io_test.py
View file

@ -1,7 +1,8 @@
import os
import gzip
import numpy as np
import pyerrors as pe import pyerrors as pe
import pyerrors.input.json as jsonio import pyerrors.input.json as jsonio
import numpy as np
import os
def test_jsonio(): def test_jsonio():
@ -70,6 +71,15 @@ def test_jsonio():
def test_json_string_reconstruction(): def test_json_string_reconstruction():
my_obs = pe.Obs([np.random.rand(100)], ['name']) my_obs = pe.Obs([np.random.rand(100)], ['name'])
json_string = pe.input.json.create_json_string(my_obs) json_string = pe.input.json.create_json_string(my_obs)
reconstructed_obs = pe.input.json.import_json_string(json_string) reconstructed_obs1 = pe.input.json.import_json_string(json_string)
assert my_obs == reconstructed_obs assert my_obs == reconstructed_obs1
compressed_string = gzip.compress(json_string.encode('utf-8'))
reconstructed_string = gzip.decompress(compressed_string).decode('utf-8')
reconstructed_obs2 = pe.input.json.import_json_string(reconstructed_string)
assert reconstructed_string == json_string
assert my_obs == reconstructed_obs2