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: tests for reweighting, merging and correlate extended

Browse source
This commit is contained in:
Fabian Joswig 2021-12-23 12:20:55 +01:00
parent 3a57471ccf
commit 2ccbd97b39
1 changed files with 25 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

25
tests/obs_test.py
View file

@ -428,6 +428,14 @@ def test_reweighting():
assert r_obs[0].reweighted assert r_obs[0].reweighted
r_obs2 = r_obs[0] * my_obs r_obs2 = r_obs[0] * my_obs
assert r_obs2.reweighted assert r_obs2.reweighted
my_covobs = pe.cov_Obs(1.0, 0.003, 'cov')
with pytest.raises(Exception):
pe.reweight(my_obs, [my_covobs])
my_obs2 = pe.Obs([np.random.rand(1000)], ['t2'])
with pytest.raises(Exception):
pe.reweight(my_obs, [my_obs + my_obs2])
with pytest.raises(Exception):
pe.reweight(my_irregular_obs, [my_obs])
def test_merge_obs(): def test_merge_obs():
@ -436,6 +444,12 @@ def test_merge_obs():
merged = pe.merge_obs([my_obs1, my_obs2]) merged = pe.merge_obs([my_obs1, my_obs2])
diff = merged - my_obs2 - my_obs1 diff = merged - my_obs2 - my_obs1
assert diff == -(my_obs1.value + my_obs2.value) / 2 assert diff == -(my_obs1.value + my_obs2.value) / 2
with pytest.raises(Exception):
pe.merge_obs([my_obs1, my_obs1])
my_covobs = pe.cov_Obs(1.0, 0.003, 'cov')
with pytest.raises(Exception):
pe.merge_obs([my_obs1, my_covobs])
def test_merge_obs_r_values(): def test_merge_obs_r_values():
@ -468,6 +482,17 @@ def test_correlate():
corr3 = pe.correlate(my_obs5, my_obs6) corr3 = pe.correlate(my_obs5, my_obs6)
assert my_obs5.idl == corr3.idl assert my_obs5.idl == corr3.idl
my_new_obs = pe.Obs([np.random.rand(100)], ['q3'])
with pytest.raises(Exception):
pe.correlate(my_obs1, my_new_obs)
my_covobs = pe.cov_Obs(1.0, 0.003, 'cov')
with pytest.raises(Exception):
pe.correlate(my_covobs, my_covobs)
r_obs = pe.reweight(my_obs1, [my_obs1])[0]
with pytest.warns(RuntimeWarning):
pe.correlate(r_obs, r_obs)
def test_irregular_error_propagation(): def test_irregular_error_propagation():
obs_list = [pe.Obs([np.random.rand(100)], ['t']), obs_list = [pe.Obs([np.random.rand(100)], ['t']),