fix: CObs can now be added and multiplied to as well as subtracted from

Obs in all combinations
This commit is contained in:
Fabian Joswig 2022-02-08 17:07:40 +00:00
parent f923ad06f7
commit f51503555b
2 changed files with 29 additions and 7 deletions

View file

@ -117,6 +117,10 @@ def test_function_overloading():
np.arctanh(1 / b)
np.sinc(1 / b)
b ** b
0.5 ** b
b ** 0.5
def test_overloading_vectorization():
a = np.random.randint(1, 100, 10)
@ -392,6 +396,9 @@ def test_cobs():
obs2 = pe.pseudo_Obs(-0.2, 0.03, 't')
my_cobs = pe.CObs(obs1, obs2)
my_cobs == my_cobs
str(my_cobs)
repr(my_cobs)
assert not (my_cobs + my_cobs.conjugate()).real.is_zero()
assert (my_cobs + my_cobs.conjugate()).imag.is_zero()
assert (my_cobs - my_cobs.conjugate()).real.is_zero()
@ -424,6 +431,23 @@ def test_cobs():
assert (other / my_cobs * my_cobs - other).is_zero()
def test_cobs_overloading():
obs = pe.pseudo_Obs(1.1, 0.1, 't')
cobs = pe.CObs(obs, obs)
cobs + obs
obs + cobs
cobs - obs
obs - cobs
cobs * obs
obs * cobs
cobs / obs
obs / cobs
def test_reweighting():
my_obs = pe.Obs([np.random.rand(1000)], ['t'])
assert not my_obs.reweighted