From 7682e2914605007e9c0253337278630ae13ef3c1 Mon Sep 17 00:00:00 2001 From: Fabian Joswig Date: Fri, 20 Jan 2023 11:54:43 +0000 Subject: [PATCH] test: tests refactored. --- tests/obs_test.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tests/obs_test.py b/tests/obs_test.py index 6c4817b7..cc0203c1 100644 --- a/tests/obs_test.py +++ b/tests/obs_test.py @@ -8,7 +8,9 @@ from hypothesis import given, strategies as st np.random.seed(0) -@given(st.lists(st.floats(allow_nan=False, allow_infinity=False, width=32), min_size=5), st.text(), st.floats(allow_nan=False, allow_infinity=False, width=32, min_value=0)) +@given(st.lists(st.floats(allow_nan=False, allow_infinity=False, width=32), min_size=5), + st.text(), + st.floats(allow_nan=False, allow_infinity=False, width=32, min_value=0)) def test_fuzzy_obs(data, string, S): my_obs = pe.Obs([data], [string]) my_obs * my_obs @@ -66,7 +68,7 @@ def test_Obs_exceptions(): one.plot_piechart() plt.close('all') -def test_dump(): +def test_dump_pickle(): value = np.random.normal(5, 10) dvalue = np.abs(np.random.normal(0, 1)) test_obs = pe.pseudo_Obs(value, dvalue, 't') @@ -75,6 +77,12 @@ def test_dump(): new_obs = pe.load_object('test_dump.p') os.remove('test_dump.p') assert test_obs == new_obs + + +def test_dump_json(): + value = np.random.normal(5, 10) + dvalue = np.abs(np.random.normal(0, 1)) + test_obs = pe.pseudo_Obs(value, dvalue, 't') test_obs.dump('test_dump', dataype="json.gz", path=".") test_obs.dump('test_dump', dataype="json.gz") new_obs = pe.input.json.load_json("test_dump")