def test_shooting_move(self):
     self.shooter = ForwardShootMover(ensemble=self.ens,
                                      selector=FirstFrameSelector(),
                                      engine=self.dyn)
     change = self.shooter.move(self.gs)
     samples = change.trials
     assert len(samples) == 1
     assert change.accepted is True
     truth = [-0.5, -0.4, -0.3, -0.2, -0.1]
     assert truth == [s.coordinates[0][0] for s in samples[0].trajectory]
Esempio n. 2
0
 def test_shooting_move(self):
     self.shooter = ForwardShootMover(ensemble=self.ens,
                                      selector=FirstFrameSelector(),
                                      engine=self.dyn)
     change = self.shooter.move(self.gs)
     samples = change.trials
     assert_equal(len(samples), 1)
     assert_equal(change.accepted, True)
     assert_items_equal(
         [-0.5, -0.4, -0.3, -0.2, -0.1],
         [s.coordinates[0][0] for s in samples[0].trajectory])