def test_total_state_postreset(self): testInputs = [2, 3, 4, 5, 6, 10, 999] for testInput in testInputs: gs = GameStarter(testInput, 1.0, 2.0, 0.5) gs.push(1) gs.timeStep(1.5) self.assertEqual(1, gs.totalInState("ACTIVE")) self.assertEqual(testInput-1, gs.totalInState("OUT")) gs.resetAll() self.assertEqual(testInput, gs.totalInState("OUT"))