mirror of
https://github.com/fjosw/pyerrors.git
synced 2025-03-15 23:00:25 +01:00
49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import pyerrors.obs as pe
|
|
import pyerrors.input.json as jsonio
|
|
import numpy as np
|
|
import os
|
|
|
|
|
|
def test_jsonio():
|
|
o = pe.pseudo_Obs(1.0, .2, 'one')
|
|
o2 = pe.pseudo_Obs(0.5, .1, 'two|r1')
|
|
o3 = pe.pseudo_Obs(0.5, .1, 'two|r2')
|
|
o4 = pe.merge_obs([o2, o3])
|
|
otag = 'This has been merged!'
|
|
o4.tag = otag
|
|
do = o - .2 * o4
|
|
|
|
o5 = pe.pseudo_Obs(0.8, .1, 'two|r2')
|
|
testl = [o3, o5]
|
|
|
|
arr = np.array([o3, o5])
|
|
mat = np.array([[pe.pseudo_Obs(1.0, .1, 'mat'), pe.pseudo_Obs(0.3, .1, 'mat')], [pe.pseudo_Obs(0.2, .1, 'mat'), pe.pseudo_Obs(2.0, .4, 'mat')]])
|
|
|
|
ol = [o4, do, testl, mat, arr, np.array([o])]
|
|
fname = 'test_rw'
|
|
|
|
jsonio.dump_to_json(ol, fname, indent=1)
|
|
|
|
rl = jsonio.load_json(fname)
|
|
|
|
os.remove(fname + '.json.gz')
|
|
|
|
for i in range(len(rl)):
|
|
if isinstance(ol[i], pe.Obs):
|
|
o = ol[i] - rl[i]
|
|
assert(o.is_zero())
|
|
assert(ol[i].tag == rl[i].tag)
|
|
or1 = np.ravel(ol[i])
|
|
or2 = np.ravel(rl[i])
|
|
for j in range(len(or1)):
|
|
o = or1[j] - or2[j]
|
|
assert(o.is_zero())
|
|
|
|
description = {'I': {'Am': {'a': 'nested dictionary!'}}}
|
|
jsonio.dump_to_json(ol, fname, indent=1, gz=False, description=description)
|
|
|
|
rl = jsonio.load_json(fname, gz=False, full_output=True)
|
|
|
|
assert(description == rl['description'])
|
|
|
|
os.remove(fname + '.json')
|