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))