def test_horse_attack_king(self):
board = [
# 0 1 2 3
[None, None, None, None], # 0
[None, None, None, None], # 1
[None, 'kw', None, None], # 2
[None, 'hb', None, None], # 3
]
is_valid = king(2, 1, 1, 2, board)
self.assertEqual(is_valid, False)
def test_attack_top_right(self):
board = [
# 0 1 2 3
[None, None, None, None], # 0
[None, None, None, None], # 1
[None, 'kw', None, None], # 2
[None, None, None, None], # 3
]
is_valid = king(2, 1, 1, 2, board)
self.assertEqual(is_valid, True)
def test_attack_bottom_left(self):
board = [
# 0 1 2 3
[None, None, None, None], # 0
[None, None, None, None], # 1
[None, 'kw', None, None], # 2
[None, None, None, None], # 3
]
is_valid = king(2, 1, 3, 0, board)
self.assertEqual(is_valid, True)
def test_king_can_attack_enemy(self):
board = [
# 0 1 2 3
[None, None, None, None], # 0
[None, None, 'qw', None], # 1
[None, 'kb', None, None], # 2
[None, None, None, None], # 3
]
is_valid = king(2, 1, 1, 2, board)
self.assertEqual(is_valid, True)
def test_king_cannot_move_to_attacking_place(self):
board = [
# 0 1 2 3
['hw', None, None, None], # 0
[None, None, None, None], # 1
[None, 'kb', None, None], # 2
[None, None, None, None], # 3
]
is_valid = king(2, 1, 1, 2, board)
self.assertEqual(is_valid, False)
def test_one_of_two_can_attack_king(self):
board = [
# 0 1 2 3
[None, None, None, None], # 0
[None, None, None, None], # 1
[None, 'kb', None, None], # 2
[None, 'hw', None, 'pw'], # 3
]
is_valid = king(2, 1, 1, 2, board)
self.assertEqual(is_valid, False)
