feat: hash method for Obs objects added.

This commit is contained in:
Fabian Joswig 2022-07-01 16:53:08 +01:00
parent f908b6328d
commit b2d5263ea3
2 changed files with 19 additions and 0 deletions

View file

@ -684,6 +684,12 @@ class Obs:
else:
return '{:.0f}({:2.0f})'.format(self.value, self._dvalue)
def __hash__(self):
return hash((np.array([self.value]).astype(np.float32)[0],
tuple([o.astype(np.float32).data.tobytes() for o in self.deltas.values()]),
tuple(np.array([o.errsq() for o in self.covobs.values()]).astype(np.float32).data.tobytes()),
tuple(self.names)))
# Overload comparisons
def __lt__(self, other):
return self.value < other

View file

@ -950,3 +950,16 @@ def test_cobs_array():
cobs * np.identity(4)
np.identity(4) / cobs
cobs / np.ones((4, 4))
def test_hash():
obs = pe.pseudo_Obs(0.3, 0.1, "test") + pe.Obs([np.random.normal(2.3, 0.2, 200)], ["test2"], [range(1, 400, 2)])
o1 = obs + pe.cov_Obs(0.0, 0.1, "co") + pe.cov_Obs(0.0, 0.8, "co2")
o2 = obs + pe.cov_Obs(0.0, 0.2, "co") + pe.cov_Obs(0.0, 0.8, "co2")
for i_obs in [obs, o1, o2]:
assert hash(i_obs) == hash(i_obs ** 2 / i_obs) == hash(1 * i_obs)
assert hash(i_obs) == hash((1 + 1e-16) * i_obs)
assert hash(i_obs) != hash((1 + 1e-7) * i_obs)
assert hash(obs) != hash(o1)
assert hash(o1) != hash(o2)