def testGeneric(self):
physics = fake_physics.FakePhysics()
repeated_observable = base.Generic(fake_physics.FakePhysics.repeated,
update_interval=42)
repeated_observation = repeated_observable.observation_callable(
physics)()
self.assertEqual(repeated_observable.update_interval, 42)
np.testing.assert_array_equal(repeated_observation, [0, 0])
def testCorruptor(self):
physics = fake_physics.FakePhysics()
def add_twelve(old_value, random_state):
del random_state # Unused.
return [x + 12 for x in old_value]
repeated_observable = base.Generic(fake_physics.FakePhysics.repeated,
corruptor=add_twelve)
corrupted = repeated_observable.observation_callable(
physics=physics, random_state=None)()
np.testing.assert_array_equal(corrupted, [12, 12])
def observable1(self):
return observable.Generic(lambda phys: 1.0)
def observable0(self):
return observable.Generic(lambda phys: 0.0)
