def create_pieces(self, square_side_size):
radius = square_side_size * 0.9 / 2
if self.player_nr == 2:
for i in range(1, 7):
if i == 2 or i == 5:
piece = Piece(i * square_side_size + square_side_size / 2, square_side_size / 2, 'v', radius)
self.pieces[piece.get_position()] = piece
piece = Piece(i * square_side_size + square_side_size / 2,
7 * square_side_size + square_side_size / 2,
'v', radius)
self.pieces[piece.get_position()] = piece
else:
piece = Piece(square_side_size / 2, i * square_side_size + square_side_size / 2, 'h', radius)
self.pieces[piece.get_position()] = piece
piece = Piece(7 * square_side_size + square_side_size / 2,
i * square_side_size + square_side_size / 2,
'h', radius)
self.pieces[piece.get_position()] = piece
else:
for i in range(1, 7):
if i == 2 or i == 5:
piece = Piece(square_side_size / 2, i * square_side_size + square_side_size / 2, 'h', radius)
self.pieces[piece.get_position()] = piece
piece = Piece(7 * square_side_size + square_side_size / 2,
i * square_side_size + square_side_size / 2,
'h', radius)
self.pieces[piece.get_position()] = piece
else:
piece = Piece(i * square_side_size + square_side_size / 2, square_side_size / 2, 'v', radius)
self.pieces[piece.get_position()] = piece
piece = Piece(i * square_side_size + square_side_size / 2,
7 * square_side_size + square_side_size / 2,
'v', radius)
self.pieces[piece.get_position()] = piece
def test_piece_getters():
test_piece = Piece('16WN')
assert test_piece.get_position() == '16'
assert test_piece.get_color() == 'W'
assert test_piece.is_king() == False
def test_vul_pos_right(gamestate):
print("Testing vulnerable pos right")
piece1 = Piece(300, 60, "h", 50)
piece2 = Piece(180, 60, "h", 50)
piece25 = Piece(420, 60, "v", 50)
piece3 = Piece(420, 60, "h", 50)
# gamestate.players[1].pieces[piece3.get_position()] = piece3
# Test 1st Return
print("Test 1st Return")
gamestate.players[0].pieces = defaultdict(Piece)
gamestate.players[1].pieces = defaultdict(Piece)
gamestate.players[0].pieces[piece1.get_position()] = piece1
gamestate.players[0].pieces[
piece2.get_position()] = piece2 # Not Vulnerable
print("P1: ", gamestate.players[0].pieces)
print("P2: ", gamestate.players[1].pieces)
print(heuristics.vuln_pos_right(gamestate, 180, 60, 0, 1))
# Test 2nd Return
print("Test 2nd Return")
gamestate.players[0].pieces = defaultdict(Piece)
gamestate.players[1].pieces = defaultdict(Piece)
gamestate.players[0].pieces[piece2.get_position()] = piece2 # vulnerable
piece3.evolve()
gamestate.players[1].pieces[piece3.get_position()] = piece3
print("P1: ", gamestate.players[0].pieces)
print("P2: ", gamestate.players[1].pieces)
print(heuristics.vuln_pos_right(gamestate, 180, 60, 0, 1))
# Test 3rd Return
print("Test 3rd Return")
gamestate.players[0].pieces = defaultdict(Piece)
gamestate.players[1].pieces = defaultdict(Piece)
gamestate.players[0].pieces[piece3.get_position()] = piece3 # vulnerable
gamestate.players[1].pieces[piece1.get_position()] = piece1
print("P1: ", gamestate.players[0].pieces)
print("P2: ", gamestate.players[1].pieces)
print(heuristics.vuln_pos_right(gamestate, 300, 60, 1, 0))
# Test 4th Return
print("Test 4th Return")
gamestate.players[0].pieces = defaultdict(Piece)
gamestate.players[1].pieces = defaultdict(Piece)
gamestate.players[0].pieces[
piece1.get_position()] = piece1 # Not vulnerable
gamestate.players[1].pieces[piece25.get_position()] = piece25
print("P1: ", gamestate.players[0].pieces)
print("P2: ", gamestate.players[1].pieces)
print(heuristics.vuln_pos_right(gamestate, 300, 60, 0, 1))
# Test Last Return
print("Test 5th Return")
gamestate.players[0].pieces = defaultdict(Piece)
gamestate.players[1].pieces = defaultdict(Piece)
gamestate.players[0].pieces[
piece1.get_position()] = piece1 # Not vulnerable
# gamestate.players[1].pieces[piece2.get_position()] = piece2
print("P1: ", gamestate.players[0].pieces)
print("P2: ", gamestate.players[1].pieces)
print(heuristics.vuln_pos_right(gamestate, 300, 60, 0, 1))
