def test_pause_resume(self): # Initialisation step_d, step = simple_step() expt = Experiment(step) expt_d = expt.run() # Pause pause_d = expt.pause() step.pause.assert_called_once_with() self.assertEqual(expt.state, State.PAUSED) self.assertEqual(pause_d.result, [None]) # Resume resume_d = expt.resume() step.resume.assert_called_once_with() self.assertEqual(expt.state, State.RUNNING) self.assertEqual(resume_d.result, [None]) step_d.callback(None)
def test_pause_resume (self): # Initialisation step_d, step = simple_step() expt = Experiment(step) expt_d = expt.run() # Pause pause_d = expt.pause() step.pause.assert_called_once_with() self.assertEqual(expt.state, State.PAUSED) self.assertEqual(pause_d.result, [None]) # Resume resume_d = expt.resume() step.resume.assert_called_once_with() self.assertEqual(expt.state, State.RUNNING) self.assertEqual(resume_d.result, [None]) step_d.callback(None)