def test_king():
current_square = Square('e', '4')
piece = Queen(Colour.WHITE)
current_square.set_piece(piece)
to_square = Square('e', '5')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('e', '3')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('f', '3')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('f', '4')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('f', '5')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('d', '3')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('d', '4')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('d', '5')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('d', '6')
assert not piece.valid_change(to_square, is_capture=False)
def test_queen():
current_square = Square('e', '4')
piece = Queen(Colour.WHITE)
current_square.set_piece(piece)
to_square = Square('h', '1')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('e', '1')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('a', '4')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('h', '4')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('h', '1')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('b', '1')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('a', '8')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('h', '7')
assert piece.valid_change(to_square, is_capture=False)
to_square = Square('a', '5')
assert not piece.valid_change(to_square, is_capture=False)
