87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
from diplomacy import Game
|
|
|
|
def test_nmr():
|
|
game = Game()
|
|
# France does nothing
|
|
# England moves
|
|
game.set_orders("ENGLAND", ["F LON - ENG", "A LVP - WAL", "F EDI - NTH"])
|
|
|
|
print("Processing Spring 1901...")
|
|
game.process()
|
|
|
|
# Check France
|
|
france = game.get_power("FRANCE")
|
|
print(f"France Civil Disorder Status: {france.civil_disorder}")
|
|
print(f"France Units: {france.units}")
|
|
# France orders in the previous phase (Spring 1901)
|
|
# The game object stores history?
|
|
# game.get_orders(power_name) gets *current* phase orders (empty)
|
|
|
|
# We can inspect the map or units to see if they moved?
|
|
# But France didn't move.
|
|
|
|
# Let's verify they are still at starting positions
|
|
print(f"France Units after process: {france.units}")
|
|
assert 'A PAR' in france.units
|
|
assert 'F BRE' in france.units
|
|
assert 'A MAR' in france.units
|
|
|
|
# Also Check if we can find the 'H' orders in history if available
|
|
# game.order_history is a thing?
|
|
# From dir(game) in previous turn: 'order_history'
|
|
|
|
# keys of order_history are timestamps or phase names?
|
|
# Let's print keys
|
|
# print(f"Order History Keys: {game.order_history.keys()}")
|
|
# It might be keyed by phase name 'SPRING 1901 MOVEMENT'
|
|
|
|
|
|
# Test 2: Auto-dsiband in Adjustment Phase
|
|
print("\nTesting Auto-Disband...")
|
|
# Create a situation where France loses a center and must disband
|
|
# We can force set units and centers
|
|
g2 = Game()
|
|
g2.set_units("FRANCE", ["A BUR", "A PAR"]) # 2 units
|
|
g2.set_centers("FRANCE", ["PAR"]) # 1 center (needs to remove 1)
|
|
# Phase needs to be adjustment? Or just process?
|
|
# Game starts at Spring 1901. We need to set phase to Winter 1901?
|
|
# Or just manipulate state to be in adjustment?
|
|
|
|
# Easiest way: force phase
|
|
from diplomacy.utils.game_phase_data import GamePhaseData
|
|
g2.phase = "WINTER 1901 ADJUSTMENTS"
|
|
|
|
# We must ensure the game knows it's adjustment phase data structure
|
|
# Actually just processing might not work if we brute force phase string?
|
|
# Better to correct way:
|
|
# Let's just use a map where we can skip to adjustments or manually set it up correct.
|
|
# brute forcing might break internal state.
|
|
|
|
# Alternative:
|
|
# 1. France holds
|
|
# 2. Enemy takes a center
|
|
# 3. Fall ends
|
|
# 4. France must disband
|
|
|
|
# Let's try brute forcing phase + clear cache if possible, but let's try the cleaner way
|
|
# Set Units/Centers then process?
|
|
# The game checks phase validity.
|
|
|
|
# Let's try:
|
|
g3 = Game()
|
|
g3.set_units("FRANCE", ["A PAR", "A MAR"])
|
|
g3.set_centers("FRANCE", ["PAR"]) # Lost MAR
|
|
g3.phase = "WINTER 1901 ADJUSTMENTS"
|
|
# We might need to ensure 'A MAR' is considered a unit.
|
|
|
|
print("Processing Adjustment Phase (No Orders for France)...")
|
|
g3.process()
|
|
print(f"France Units after adjustment: {g3.get_units('FRANCE')}")
|
|
# A MAR is further from PAR than A PAR?
|
|
# Distance PAR-PAR = 0.
|
|
# Distance MAR-PAR = 1.
|
|
# So A MAR should be disbanded.
|
|
|
|
if __name__ == "__main__":
|
|
test_nmr()
|