fix: Corr.symmetric can now deal with None entries. (#145)

This commit is contained in:
Fabian Joswig 2023-01-16 16:10:19 +00:00 committed by GitHub
parent 26447d658c
commit 88fd37b241
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 2 deletions

View file

@ -552,3 +552,17 @@ def test_corr_no_filtering():
b = pe.pseudo_Obs(1, 1e-11, 'a', samples=30)
c *= b
assert np.all([c[0].idl == o.idl for o in c])
def test_corr_symmetric():
obs = []
for _ in range(4):
obs.append(pe.pseudo_Obs(np.random.rand(), 0.1, "test"))
for corr in [pe.Corr([obs[0] + 8, obs[1], obs[2], obs[3]]),
pe.Corr([obs[0] + 8, obs[1], obs[2], None]),
pe.Corr([None, obs[1], obs[2], obs[3]])]:
scorr = corr.symmetric()
assert scorr[1] == scorr[3]
assert scorr[2] == corr[2]
assert scorr[0] == corr[0]