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)