def test_find_ship_vertical(self):
base = SeaField(5, 5)
SeaPlayground.put_ship(base, 1, 1, 3, True)
assert base.find_ship_by_cells(1, 2) == {(1, 1), (1, 2), (1, 3)}
def test_find_abcent(self):
base = SeaField(5, 5)
assert base.find_ship_by_cells(1, 2) == set()
def test_find_ship(self):
base = SeaField(5, 5)
SeaPlayground.put_ship(base, 1, 1, 3)
assert base.find_ship_by_cells(2, 1) == {(1, 1), (2, 1), (3, 1)}