def test_remove_from_irregular_gen():
squares = [(0, 0), (0, 1), (1, 1), (1, 2), (1, 3)]
def gen_sq():
for sq in squares:
yield sq
board = Irregular(gen_sq())
assert set(board.squares) == set(squares)
trimmed = board.remove((1, 3))
assert set(trimmed.squares) == set([(0, 0), (0, 1), (1, 1), (1, 2)])
def test_remove_from_irregular():
squares = [(0, 0), (0, 1), (1, 1), (1, 2), (1, 3)]
board = Irregular(squares)
assert set(board.squares) == set(squares)
trimmed = board.remove((1, 3))
assert set(trimmed.squares) == set([(0, 0), (0, 1), (1, 1), (1, 2)])
