def test_find_koma(self):
shogi = Shogi()
koma_positions = shogi.find_koma(Koma.kin)
self.assertIn([3, 8], koma_positions)
self.assertIn([5, 8], koma_positions)
koma_positions = shogi.find_koma(Koma.opponent_kin)
self.assertIn([3, 0], koma_positions)
self.assertIn([5, 0], koma_positions)
koma_positions = shogi.find_koma(Koma.hisha)
self.assertIn([7, 7], koma_positions)
def test_find_koma(self):
shogi = Shogi()
koma_positions = shogi.find_koma(Koma.kin)
self.assertIn([3, 8], koma_positions)
self.assertIn([5, 8], koma_positions)
koma_positions = shogi.find_koma(Koma.opponent_kin)
self.assertIn([3, 0], koma_positions)
self.assertIn([5, 0], koma_positions)
koma_positions = shogi.find_koma(Koma.hisha)
self.assertIn([7, 7], koma_positions)
def test_drop(self):
shogi = Shogi()
shogi.first_tegoma = [Koma.kin]
shogi.second_tegoma = [Koma.opponent_kin]
shogi.drop(Koma.kin, 4, 4)
self.assertEqual(shogi.board[4][4], Koma.kin)
shogi.drop(Koma.opponent_kin, 3, 3)
self.assertEqual(shogi.board[3][3], Koma.opponent_kin)
def test_movable_for_kyo(self):
shogi = Shogi()
shogi.move(8, 6, 0, 0, False)
shogi.first = True
movable = shogi.movable(8, 8, 8, 4, False)
self.assertTrue(movable)
movable = shogi.movable(8, 8, 8, 3, False)
self.assertTrue(movable)
movable = shogi.movable(8, 8, 8, 2, False)
self.assertTrue(movable)
movable = shogi.movable(8, 8, 8, 1, False)
self.assertFalse(movable)
def test_droppable_nifu(self):
shogi = Shogi()
shogi.first_tegoma = [Koma.fu]
shogi.second_tegoma = [Koma.opponent_fu]
shogi.first = True
droppable = shogi.droppable(Koma.fu, 4, 4)
self.assertFalse(droppable)
shogi.first = False
droppable = shogi.droppable(Koma.opponent_fu, 4, 4)
self.assertFalse(droppable)
def test_drop(self):
shogi = Shogi()
shogi.first_tegoma = [Koma.kin]
shogi.second_tegoma = [Koma.opponent_kin]
shogi.drop(Koma.kin, 4, 4)
self.assertEqual(shogi.board[4][4], Koma.kin)
shogi.drop(Koma.opponent_kin, 3, 3)
self.assertEqual(shogi.board[3][3], Koma.opponent_kin)
def test_movable_for_kyo(self):
shogi = Shogi()
shogi.move(8, 6, 0, 0, False)
shogi.first = True
movable = shogi.movable(8, 8, 8, 4, False)
self.assertTrue(movable)
movable = shogi.movable(8, 8, 8, 3, False)
self.assertTrue(movable)
movable = shogi.movable(8, 8, 8, 2, False)
self.assertTrue(movable)
movable = shogi.movable(8, 8, 8, 1, False)
self.assertFalse(movable)
def __init__(self, channel_id, users, validator=BasicUserValidator()):
self._shogi = ShogiModule()
self.channel_id = channel_id
self.user_ids = [x["id"] for x in users]
random.shuffle(users)
self.first_user = ShogiUser(users[0]["id"], users[0]["name"])
self.second_user = ShogiUser(users[1]["id"], users[1]["name"])
self.id = uuid.uuid4().hex
self._validator = validator
self.kifu = Kifu()
def __init__(self, channel_id, users, validator=BasicUserValidator()):
self._shogi = ShogiModule()
self.channel_id = channel_id
self.user_ids = [x["id"] for x in users]
random.shuffle(users)
self.first_user = ShogiUser(users[0]["id"], users[0]["name"])
self.second_user = ShogiUser(users[1]["id"], users[1]["name"])
self.id = uuid.uuid4().hex
self._validator = validator
self.kifu = Kifu()
def test_last_move(self):
shogi = Shogi()
shogi.second_tegoma = [Koma.opponent_fu]
shogi.move(0, 6, 0, 5, False)
self.assertEqual(shogi.last_move_x, 0)
self.assertEqual(shogi.last_move_y, 5)
shogi.drop(Koma.opponent_fu, 5, 5)
self.assertEqual(shogi.last_move_x, 5)
self.assertEqual(shogi.last_move_y, 5)
def test_movable_for_enemy(self):
shogi = Shogi()
movable = shogi.movable(0, 0, 0, 1, False)
self.assertFalse(movable)
movable = shogi.movable(0, 1, 0, 2, False)
self.assertFalse(movable)
shogi.move(2, 6, 2, 5, False)
movable = shogi.movable(0, 0, 0, 1, False)
self.assertTrue(movable)
def matta(self):
if len(self.kifu.kifu) == 0:
raise KomaCannotMoveException
self.kifu.pop()
self._shogi = ShogiModule()
for kifu in self.kifu.kifu:
from_x, from_y, to_x, to_y, promote, koma = kifu
if koma is None:
self._shogi.move(from_x, from_y, to_x, to_y, promote)
else:
self._shogi.drop(koma, to_x, to_y)
def matta(self):
if len(self.kifu.kifu) == 0:
raise KomaCannotMoveException
self.kifu.pop()
self._shogi = ShogiModule()
for kifu in self.kifu.kifu:
from_x, from_y, to_x, to_y, promote, koma = kifu
if koma is None:
self._shogi.move(from_x, from_y, to_x, to_y, promote)
else:
self._shogi.drop(koma, to_x, to_y)
def test_droppable_nifu(self):
shogi = Shogi()
shogi.first_tegoma = [Koma.fu]
shogi.second_tegoma = [Koma.opponent_fu]
shogi.first = True
droppable = shogi.droppable(Koma.fu, 4, 4)
self.assertFalse(droppable)
shogi.first = False
droppable = shogi.droppable(Koma.opponent_fu, 4, 4)
self.assertFalse(droppable)
def test_movable_for_enemy(self):
shogi = Shogi()
movable = shogi.movable(0, 0, 0, 1, False)
self.assertFalse(movable)
movable = shogi.movable(0, 1, 0, 2, False)
self.assertFalse(movable)
shogi.move(2, 6, 2, 5, False)
movable = shogi.movable(0, 0, 0, 1, False)
self.assertTrue(movable)
def test_last_move(self):
shogi = Shogi()
shogi.second_tegoma = [Koma.opponent_fu]
shogi.move(0, 6, 0, 5, False)
self.assertEqual(shogi.last_move_x, 0)
self.assertEqual(shogi.last_move_y, 5)
shogi.drop(Koma.opponent_fu, 5, 5)
self.assertEqual(shogi.last_move_x, 5)
self.assertEqual(shogi.last_move_y, 5)
def test_droppable(self):
shogi = Shogi()
shogi.first_tegoma = [Koma.kin]
shogi.second_tegoma = [Koma.opponent_kin]
shogi.first = True
# have
droppable = shogi.droppable(Koma.kin, 4, 4)
self.assertTrue(droppable)
# opponent
droppable = shogi.droppable(Koma.opponent_kin, 4, 4)
self.assertFalse(droppable)
# not have
droppable = shogi.droppable(Koma.gin, 4, 4)
self.assertFalse(droppable)
shogi.first = False
droppable = shogi.droppable(Koma.opponent_kin, 4, 4)
self.assertTrue(droppable)
# opponent
droppable = shogi.droppable(Koma.kin, 4, 4)
self.assertFalse(droppable)
# not have
droppable = shogi.droppable(Koma.opponent_gin, 4, 4)
self.assertFalse(droppable)
class Shogi:
def __init__(self, channel_id, users, validator=BasicUserValidator()):
self._shogi = ShogiModule()
self.channel_id = channel_id
self.user_ids = [x["id"] for x in users]
random.shuffle(users)
self.first_user = ShogiUser(users[0]["id"], users[0]["name"])
self.second_user = ShogiUser(users[1]["id"], users[1]["name"])
self.id = uuid.uuid4().hex
self._validator = validator
self.kifu = Kifu()
def move(self, from_x, from_y, to_x, to_y, promote):
self._shogi.move(from_x, from_y, to_x, to_y, promote)
self.kifu.add(from_x, from_y, to_x, to_y, promote, None)
def drop(self, koma, to_x, to_y):
self._shogi.drop(koma, to_x, to_y)
self.kifu.add(-1, -1, to_x, to_y, False, koma)
def movable(self, from_x, from_y, to_x, to_y, promote):
return self._shogi.movable(from_x, from_y, to_x, to_y, promote)
def droppable(self, koma, to_x, to_y):
return self._shogi.droppable(koma, to_x, to_y)
def find_koma(self, koma):
return self._shogi.find_koma(koma)
def validate(self, shogi, user_id):
return self._validator.validate(shogi, user_id)
def set_validator(self, validator):
self._validator = validator
def matta(self):
if len(self.kifu.kifu) == 0:
raise KomaCannotMoveException
self.kifu.pop()
self._shogi = ShogiModule()
for kifu in self.kifu.kifu:
from_x, from_y, to_x, to_y, promote, koma = kifu
if koma is None:
self._shogi.move(from_x, from_y, to_x, to_y, promote)
else:
self._shogi.drop(koma, to_x, to_y)
@property
def first(self):
return self._shogi.first
@property
def board(self):
return self._shogi.board
@property
def last_move_x(self):
return self._shogi.last_move_x
@property
def last_move_y(self):
return self._shogi.last_move_y
@property
def first_tegoma(self):
return self._shogi.first_tegoma
@property
def second_tegoma(self):
return self._shogi.second_tegoma
def test_move_76fu(self):
shogi = Shogi()
shogi.move(2, 6, 2, 5, False)
self.assertEqual(shogi.board[5][2], Koma.fu)
self.assertEqual(shogi.board[6][2], Koma.empty)
def test_movable_for_kaku(self):
shogi = Shogi()
movable = shogi.movable(1, 7, 3, 5, False)
self.assertFalse(movable)
# 66 fu
movable = shogi.movable(2, 6, 2, 5, False)
self.assertTrue(movable)
shogi.move(2, 6, 2, 5, False)
# 34 fu
movable = shogi.movable(6, 2, 6, 3, False)
self.assertTrue(movable)
shogi.move(6, 2, 6, 3, False)
movable = shogi.movable(1, 7, 8, 0, False)
self.assertFalse(movable)
# 22 kaku
movable = shogi.movable(1, 7, 7, 1, False)
self.assertTrue(movable)
shogi.move(6, 2, 6, 3, False)
def test_movable_for_fu(self):
shogi = Shogi()
movable = shogi.movable(0, 6, 0, 4, False)
self.assertFalse(movable)
movable = shogi.movable(0, 6, 0, 7, False)
self.assertFalse(movable)
movable = shogi.movable(0, 6, 1, 5, False)
self.assertFalse(movable)
movable = shogi.movable(0, 2, 0, 3, False)
self.assertFalse(movable)
# 96fu
movable = shogi.movable(0, 6, 0, 5, False)
self.assertTrue(movable)
shogi.move(0, 6, 0, 5, False)
# 94fu
movable = shogi.movable(0, 2, 0, 3, False)
self.assertTrue(movable)
shogi.move(0, 2, 0, 3, False)
# 95fu
movable = shogi.movable(0, 5, 0, 4, False)
self.assertTrue(movable)
shogi.move(0, 5, 0, 4, False)
# 95 do fu
movable = shogi.movable(0, 3, 0, 4, False)
self.assertTrue(movable)
shogi.move(0, 3, 0, 4, False)
# moving opponent koma
movable = shogi.movable(0, 4, 0, 5, False)
self.assertFalse(movable)
def test_move_for_promote(self):
shogi = Shogi()
shogi.board = [
[
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty,
Koma.promoted_fu, Koma.hisha, Koma.empty, Koma.kin
],
[
Koma.fu, Koma.kyosha, Koma.empty, Koma.empty, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.gyoku
],
[
Koma.opponent_fu, Koma.empty, Koma.empty, Koma.keima,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty
],
[
Koma.empty, Koma.empty, Koma.gin, Koma.empty, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty
],
[
Koma.empty, Koma.kyosha, Koma.opponent_kyosha, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty
],
[
Koma.empty, Koma.empty, Koma.opponent_gin, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty
],
[
Koma.fu, Koma.empty, Koma.empty, Koma.opponent_keima,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty
],
[
Koma.opponent_fu, Koma.opponent_kyosha, Koma.empty, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty,
Koma.opponent_gyoku
],
[
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty,
Koma.opponent_promoted_fu, Koma.opponent_hisha, Koma.empty,
Koma.opponent_kin
],
]
shogi.first = True
# 91 fu narazu
movable = shogi.movable(0, 1, 0, 0, False)
self.assertFalse(movable)
# 91 fu naru
movable = shogi.movable(0, 1, 0, 0, True)
self.assertTrue(movable)
shogi.first = False
# 99 fu narazu
movable = shogi.movable(0, 7, 0, 8, False)
self.assertFalse(movable)
# 99 fu naru
movable = shogi.movable(0, 7, 0, 8, True)
self.assertTrue(movable)
shogi.first = True
# 71 kei narazu
movable = shogi.movable(3, 2, 2, 0, False)
self.assertFalse(movable)
# 71 kei naru
movable = shogi.movable(3, 2, 2, 0, True)
self.assertTrue(movable)
shogi.first = False
# 79 kei narazu
movable = shogi.movable(3, 6, 2, 8, False)
self.assertFalse(movable)
# 79 kei naru
movable = shogi.movable(3, 6, 2, 8, True)
self.assertTrue(movable)
shogi.first = True
# 81 kyo narazu
movable = shogi.movable(1, 1, 1, 0, False)
self.assertFalse(movable)
# 81 kyo naru
movable = shogi.movable(1, 1, 1, 0, True)
self.assertTrue(movable)
shogi.first = False
# 89 kyo narazu
movable = shogi.movable(1, 7, 1, 8, False)
self.assertFalse(movable)
# 89 kyo naru
movable = shogi.movable(1, 7, 1, 8, True)
self.assertTrue(movable)
shogi.first = True
# 96 fu naru
movable = shogi.movable(0, 6, 0, 5, True)
self.assertFalse(movable)
# 96 fu narazu
movable = shogi.movable(0, 6, 0, 5, False)
self.assertTrue(movable)
shogi.first = False
# 94 fu naru
movable = shogi.movable(0, 2, 0, 3, True)
self.assertFalse(movable)
# 94 fu narazu
movable = shogi.movable(0, 2, 0, 3, False)
self.assertTrue(movable)
shogi.first = True
# 73 gin naru and narazu
movable = shogi.movable(2, 3, 2, 2, True)
self.assertTrue(movable)
movable = shogi.movable(2, 3, 2, 2, False)
self.assertTrue(movable)
shogi.first = False
# 77 gin naru and narazu
movable = shogi.movable(2, 5, 2, 6, True)
self.assertTrue(movable)
movable = shogi.movable(2, 5, 2, 6, False)
self.assertTrue(movable)
# kin and gyoku can't promote
shogi.first = False
movable = shogi.movable(8, 0, 8, 1, True)
self.assertFalse(movable)
movable = shogi.movable(7, 0, 7, 1, True)
self.assertFalse(movable)
shogi.first = True
movable = shogi.movable(1, 0, 1, 1, True)
self.assertFalse(movable)
movable = shogi.movable(0, 0, 0, 1, True)
self.assertFalse(movable)
# back promoteTrue
shogi.first = True
movable = shogi.movable(6, 0, 6, 5, True)
self.assertTrue(movable)
movable = shogi.movable(6, 0, 6, 7, True)
self.assertTrue(movable)
shogi.first = False
movable = shogi.movable(6, 8, 6, 5, True)
self.assertTrue(movable)
movable = shogi.movable(6, 8, 6, 1, True)
self.assertTrue(movable)
# promoted' koma try to promote
shogi.first = True
movable = shogi.movable(5, 0, 5, 1, True)
self.assertFalse(movable)
shogi.first = False
movable = shogi.movable(5, 8, 5, 7, True)
self.assertFalse(movable)
def test_droppable_fu_kyo_kei(self):
shogi = Shogi()
shogi.board[0][0] = Koma.empty
shogi.board[2][0] = Koma.empty
shogi.board[6][0] = Koma.empty
shogi.board[8][0] = Koma.empty
shogi.first_tegoma = [Koma.fu, Koma.kyosha, Koma.keima]
shogi.second_tegoma = [
Koma.opponent_fu, Koma.opponent_kyosha, Koma.opponent_keima
]
shogi.first = True
# keima
droppable = shogi.droppable(Koma.keima, 0, 1)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.keima, 0, 2)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_keima, 1, 2)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.keima, 0, 3)
self.assertTrue(droppable)
# fu
droppable = shogi.droppable(Koma.fu, 0, 0)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.fu, 0, 1)
self.assertTrue(droppable)
# kyo
droppable = shogi.droppable(Koma.kyosha, 0, 0)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.kyosha, 0, 1)
self.assertTrue(droppable)
shogi.first = False
# keima
droppable = shogi.droppable(Koma.opponent_keima, 0, 7)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.opponent_keima, 0, 6)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.keima, 1, 6)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.opponent_keima, 0, 5)
self.assertTrue(droppable)
# fu
droppable = shogi.droppable(Koma.opponent_fu, 0, 7)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_fu, 0, 8)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.opponent_fu, 0, 0)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_fu, 0, 1)
self.assertTrue(droppable)
# kyosha
droppable = shogi.droppable(Koma.opponent_kyosha, 0, 7)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_kyosha, 0, 8)
self.assertFalse(droppable)
def test_movable_for_fu(self):
shogi = Shogi()
movable = shogi.movable(0, 6, 0, 4, False)
self.assertFalse(movable)
movable = shogi.movable(0, 6, 0, 7, False)
self.assertFalse(movable)
movable = shogi.movable(0, 6, 1, 5, False)
self.assertFalse(movable)
movable = shogi.movable(0, 2, 0, 3, False)
self.assertFalse(movable)
# 96fu
movable = shogi.movable(0, 6, 0, 5, False)
self.assertTrue(movable)
shogi.move(0, 6, 0, 5, False)
# 94fu
movable = shogi.movable(0, 2, 0, 3, False)
self.assertTrue(movable)
shogi.move(0, 2, 0, 3, False)
# 95fu
movable = shogi.movable(0, 5, 0, 4, False)
self.assertTrue(movable)
shogi.move(0, 5, 0, 4, False)
# 95 do fu
movable = shogi.movable(0, 3, 0, 4, False)
self.assertTrue(movable)
shogi.move(0, 3, 0, 4, False)
# moving opponent koma
movable = shogi.movable(0, 4, 0, 5, False)
self.assertFalse(movable)
def test_droppable(self):
shogi = Shogi()
shogi.first_tegoma = [Koma.kin]
shogi.second_tegoma = [Koma.opponent_kin]
shogi.first = True
# have
droppable = shogi.droppable(Koma.kin, 4, 4)
self.assertTrue(droppable)
# opponent
droppable = shogi.droppable(Koma.opponent_kin, 4, 4)
self.assertFalse(droppable)
# not have
droppable = shogi.droppable(Koma.gin, 4, 4)
self.assertFalse(droppable)
shogi.first = False
droppable = shogi.droppable(Koma.opponent_kin, 4, 4)
self.assertTrue(droppable)
# opponent
droppable = shogi.droppable(Koma.kin, 4, 4)
self.assertFalse(droppable)
# not have
droppable = shogi.droppable(Koma.opponent_gin, 4, 4)
self.assertFalse(droppable)
def test_movable_for_empty(self):
shogi = Shogi()
movable = shogi.movable(5, 5, 5, 6, False)
self.assertFalse(movable)
def test_movable_for_kaku(self):
shogi = Shogi()
movable = shogi.movable(1, 7, 3, 5, False)
self.assertFalse(movable)
# 66 fu
movable = shogi.movable(2, 6, 2, 5, False)
self.assertTrue(movable)
shogi.move(2, 6, 2, 5, False)
# 34 fu
movable = shogi.movable(6, 2, 6, 3, False)
self.assertTrue(movable)
shogi.move(6, 2, 6, 3, False)
movable = shogi.movable(1, 7, 8, 0, False)
self.assertFalse(movable)
# 22 kaku
movable = shogi.movable(1, 7, 7, 1, False)
self.assertTrue(movable)
shogi.move(6, 2, 6, 3, False)
def test_movable_for_empty(self):
shogi = Shogi()
movable = shogi.movable(5, 5, 5, 6, False)
self.assertFalse(movable)
class Shogi:
def __init__(self, channel_id, users, validator=BasicUserValidator()):
self._shogi = ShogiModule()
self.channel_id = channel_id
self.user_ids = [x["id"] for x in users]
random.shuffle(users)
self.first_user = ShogiUser(users[0]["id"], users[0]["name"])
self.second_user = ShogiUser(users[1]["id"], users[1]["name"])
self.id = uuid.uuid4().hex
self._validator = validator
self.kifu = Kifu()
def move(self, from_x, from_y, to_x, to_y, promote):
self._shogi.move(from_x, from_y, to_x, to_y, promote)
self.kifu.add(from_x, from_y, to_x, to_y, promote, None)
def drop(self, koma, to_x, to_y):
self._shogi.drop(koma, to_x, to_y)
self.kifu.add(-1, -1, to_x, to_y, False, koma)
def movable(self, from_x, from_y, to_x, to_y, promote):
return self._shogi.movable(from_x, from_y, to_x, to_y, promote)
def droppable(self, koma, to_x, to_y):
return self._shogi.droppable(koma, to_x, to_y)
def find_koma(self, koma):
return self._shogi.find_koma(koma)
def validate(self, shogi, user_id):
return self._validator.validate(shogi, user_id)
def set_validator(self, validator):
self._validator = validator
def matta(self):
if len(self.kifu.kifu) == 0:
raise KomaCannotMoveException
self.kifu.pop()
self._shogi = ShogiModule()
for kifu in self.kifu.kifu:
from_x, from_y, to_x, to_y, promote, koma = kifu
if koma is None:
self._shogi.move(from_x, from_y, to_x, to_y, promote)
else:
self._shogi.drop(koma, to_x, to_y)
@property
def first(self):
return self._shogi.first
@property
def board(self):
return self._shogi.board
@property
def last_move_x(self):
return self._shogi.last_move_x
@property
def last_move_y(self):
return self._shogi.last_move_y
@property
def first_tegoma(self):
return self._shogi.first_tegoma
@property
def second_tegoma(self):
return self._shogi.second_tegoma
def test_droppable_fu_kyo_kei(self):
shogi = Shogi()
shogi.board[0][0] = Koma.empty
shogi.board[2][0] = Koma.empty
shogi.board[6][0] = Koma.empty
shogi.board[8][0] = Koma.empty
shogi.first_tegoma = [Koma.fu, Koma.kyosha, Koma.keima]
shogi.second_tegoma = [Koma.opponent_fu,
Koma.opponent_kyosha, Koma.opponent_keima]
shogi.first = True
# keima
droppable = shogi.droppable(Koma.keima, 0, 1)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.keima, 0, 2)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_keima, 1, 2)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.keima, 0, 3)
self.assertTrue(droppable)
# fu
droppable = shogi.droppable(Koma.fu, 0, 0)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.fu, 0, 1)
self.assertTrue(droppable)
# kyo
droppable = shogi.droppable(Koma.kyosha, 0, 0)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.kyosha, 0, 1)
self.assertTrue(droppable)
shogi.first = False
# keima
droppable = shogi.droppable(Koma.opponent_keima, 0, 7)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.opponent_keima, 0, 6)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.keima, 1, 6)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.opponent_keima, 0, 5)
self.assertTrue(droppable)
# fu
droppable = shogi.droppable(Koma.opponent_fu, 0, 7)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_fu, 0, 8)
self.assertFalse(droppable)
droppable = shogi.droppable(Koma.opponent_fu, 0, 0)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_fu, 0, 1)
self.assertTrue(droppable)
# kyosha
droppable = shogi.droppable(Koma.opponent_kyosha, 0, 7)
self.assertTrue(droppable)
droppable = shogi.droppable(Koma.opponent_kyosha, 0, 8)
self.assertFalse(droppable)
def test_move_for_promote(self):
shogi = Shogi()
shogi.board = [
[Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty,
Koma.promoted_fu, Koma.hisha, Koma.empty, Koma.kin],
[Koma.fu, Koma.kyosha, Koma.empty, Koma.empty, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.gyoku],
[Koma.opponent_fu, Koma.empty, Koma.empty, Koma.keima,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty],
[Koma.empty, Koma.empty, Koma.gin, Koma.empty, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty],
[Koma.empty, Koma.kyosha, Koma.opponent_kyosha, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty],
[Koma.empty, Koma.empty, Koma.opponent_gin, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty],
[Koma.fu, Koma.empty, Koma.empty, Koma.opponent_keima,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty],
[Koma.opponent_fu, Koma.opponent_kyosha, Koma.empty, Koma.empty,
Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.opponent_gyoku],
[Koma.empty, Koma.empty, Koma.empty, Koma.empty, Koma.empty,
Koma.opponent_promoted_fu, Koma.opponent_hisha, Koma.empty, Koma.opponent_kin],
]
shogi.first = True
# 91 fu narazu
movable = shogi.movable(0, 1, 0, 0, False)
self.assertFalse(movable)
# 91 fu naru
movable = shogi.movable(0, 1, 0, 0, True)
self.assertTrue(movable)
shogi.first = False
# 99 fu narazu
movable = shogi.movable(0, 7, 0, 8, False)
self.assertFalse(movable)
# 99 fu naru
movable = shogi.movable(0, 7, 0, 8, True)
self.assertTrue(movable)
shogi.first = True
# 71 kei narazu
movable = shogi.movable(3, 2, 2, 0, False)
self.assertFalse(movable)
# 71 kei naru
movable = shogi.movable(3, 2, 2, 0, True)
self.assertTrue(movable)
shogi.first = False
# 79 kei narazu
movable = shogi.movable(3, 6, 2, 8, False)
self.assertFalse(movable)
# 79 kei naru
movable = shogi.movable(3, 6, 2, 8, True)
self.assertTrue(movable)
shogi.first = True
# 81 kyo narazu
movable = shogi.movable(1, 1, 1, 0, False)
self.assertFalse(movable)
# 81 kyo naru
movable = shogi.movable(1, 1, 1, 0, True)
self.assertTrue(movable)
shogi.first = False
# 89 kyo narazu
movable = shogi.movable(1, 7, 1, 8, False)
self.assertFalse(movable)
# 89 kyo naru
movable = shogi.movable(1, 7, 1, 8, True)
self.assertTrue(movable)
shogi.first = True
# 96 fu naru
movable = shogi.movable(0, 6, 0, 5, True)
self.assertFalse(movable)
# 96 fu narazu
movable = shogi.movable(0, 6, 0, 5, False)
self.assertTrue(movable)
shogi.first = False
# 94 fu naru
movable = shogi.movable(0, 2, 0, 3, True)
self.assertFalse(movable)
# 94 fu narazu
movable = shogi.movable(0, 2, 0, 3, False)
self.assertTrue(movable)
shogi.first = True
# 73 gin naru and narazu
movable = shogi.movable(2, 3, 2, 2, True)
self.assertTrue(movable)
movable = shogi.movable(2, 3, 2, 2, False)
self.assertTrue(movable)
shogi.first = False
# 77 gin naru and narazu
movable = shogi.movable(2, 5, 2, 6, True)
self.assertTrue(movable)
movable = shogi.movable(2, 5, 2, 6, False)
self.assertTrue(movable)
# kin and gyoku can't promote
shogi.first = False
movable = shogi.movable(8, 0, 8, 1, True)
self.assertFalse(movable)
movable = shogi.movable(7, 0, 7, 1, True)
self.assertFalse(movable)
shogi.first = True
movable = shogi.movable(1, 0, 1, 1, True)
self.assertFalse(movable)
movable = shogi.movable(0, 0, 0, 1, True)
self.assertFalse(movable)
# back promoteTrue
shogi.first = True
movable = shogi.movable(6, 0, 6, 5, True)
self.assertTrue(movable)
movable = shogi.movable(6, 0, 6, 7, True)
self.assertTrue(movable)
shogi.first = False
movable = shogi.movable(6, 8, 6, 5, True)
self.assertTrue(movable)
movable = shogi.movable(6, 8, 6, 1, True)
self.assertTrue(movable)
# promoted' koma try to promote
shogi.first = True
movable = shogi.movable(5, 0, 5, 1, True)
self.assertFalse(movable)
shogi.first = False
movable = shogi.movable(5, 8, 5, 7, True)
self.assertFalse(movable)
def test_move_76fu(self):
shogi = Shogi()
shogi.move(2, 6, 2, 5, False)
self.assertEqual(shogi.board[5][2], Koma.fu)
self.assertEqual(shogi.board[6][2], Koma.empty)