def test_opponent_starts_and_draw(self): self.policy.play.side_effect = [(1,0,0)] state, rounds, turns = Episode(self.algo, self.policy).run(StateTest(False)) self.assertEqual(state.draw(), True) self.assertEqual((3, 3), (rounds, turns)) self.assertEqual(1, self.algo.update.call_count)