def test_simulation_not_complete(self): state = SimulatorState(2) state.add_item_to_floor('CG', 2) state.add_item_to_floor('AG', 3) state.add_item_to_floor('BG', 3) state.add_item_to_floor('AM', 3) state.add_item_to_floor('BM', 3) state.add_item_to_floor('CM', 3) self.assertFalse(state.is_complete())
def test_simulation_complete(self): state = SimulatorState(0) state.add_item_to_floor('LiG', 3) state.add_item_to_floor('LiM', 3) self.assertTrue(state.is_complete())