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