class LocationTest(unittest.TestCase):
def setUp(self):
self.loc = Location(0, 0, 3)
def test_get(self):
l = self.loc.get()
self.assertTrue(l)
def test_get_yx(self):
x = y = self.loc.size - 1
a = self.loc.get(y=y, x=x)
self.assertIsNotNone(a)
def test_search_free_position(self):
x, y = self.loc.search_free_position()
p = self.loc.get(x=x, y=y)
assert p == 0
def test_registred_players(self):
self.loc.register_player("nobus")
self.loc.register_player("bobus")
p = self.loc.get_current_players()
assert len(p) == 2
self.loc.unregister_player("popus")
p = self.loc.get_current_players()
assert len(p) == 2
self.loc.unregister_player("bobus")
p = self.loc.get_current_players()
assert len(p) == 1
