def test_available_moves_clear(self):
from gravita.ship import Ship
map = MockMap()
specs = MockSpecs(range=2)
ship = Ship(map, (1,1), specs)
available = set(map.values())
available.remove(map[1,1])
self.assertEqual(set(ship.available_moves()), available)
def test_available_moves_excludes_sectors_with_ships(self):
from gravita.ship import Ship
map = MockMap()
specs = MockSpecs(range=2)
map[0,0].ship = Ship(map, (0,0), specs)
map[0,1].ship = Ship(map, (0,1), specs)
map[1,2].ship = Ship(map, (1,2), specs)
ship = Ship(map, (1,1), specs)
self.assertEqual(set(ship.available_moves()),
set([map[1,0], map[2,0], map[2,1], map[0,2], map[2,2]]))
