def test_move_to(self): from gravita.ship import Ship map = MockMap() specs = MockSpecs(range=2) ship = Ship(map, (0,1), specs) self.assertEqual(map[2,0].ship, None) ship.move_to((2,0)) self.assertEqual(map[2,0].ship, ship) for sector in map.values(): if sector.location != (2, 0): self.assertEqual(sector.ship, None)