

def test_move_to_initial(self):
    black = Bot(0, (1, 1), 0, (0, 3))
    white = Bot(1, (6, 6), 1, (3, 6), current_pos=(1, 1))
    self.assertTrue(black.is_destroyer)
    black.current_pos = (4, 1)
    self.assertEqual(black.current_pos, (4, 1))
    self.assertTrue(black.is_harvester)
    self.assertTrue(white.is_harvester)
    black._to_initial()
    white._to_initial()
    self.assertEqual(black.current_pos, (1, 1))
    self.assertTrue(black.is_destroyer)
    self.assertEqual(white.current_pos, (6, 6))
    self.assertTrue(white.is_destroyer)

dataset/ETHPy150Open ASPP/pelita/test/test_datamodel.py TestBot.test_move_to_initial/original