215 lines
6.4 KiB
Python
215 lines
6.4 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Unit tests for action_parser.py
|
|
===============================
|
|
|
|
Tests the ActionParser's ability to extract structured actions
|
|
from various LLM response formats.
|
|
"""
|
|
|
|
import sys
|
|
from action_parser import parse_action, ActionType
|
|
|
|
|
|
def test_explicit_go_directions():
|
|
"""Test explicit 'Action: GO <direction>' format."""
|
|
# Cardinal directions
|
|
assert parse_action("Action: GO NORTH").type == ActionType.GO
|
|
assert parse_action("Action: GO NORTH").args == ("NORTH",)
|
|
|
|
assert parse_action("Action: GO SOUTH").type == ActionType.GO
|
|
assert parse_action("Action: GO SOUTH").args == ("SOUTH",)
|
|
|
|
assert parse_action("Action: GO EAST").type == ActionType.GO
|
|
assert parse_action("Action: GO EAST").args == ("EAST",)
|
|
|
|
assert parse_action("Action: GO WEST").type == ActionType.GO
|
|
assert parse_action("Action: GO WEST").args == ("WEST",)
|
|
|
|
print(" [PASS] Explicit GO directions")
|
|
|
|
|
|
def test_short_directions():
|
|
"""Test short direction abbreviations (N, S, E, W)."""
|
|
assert parse_action("Action: GO N").args == ("NORTH",)
|
|
assert parse_action("Action: GO S").args == ("SOUTH",)
|
|
assert parse_action("Action: GO E").args == ("EAST",)
|
|
assert parse_action("Action: GO W").args == ("WEST",)
|
|
|
|
print(" [PASS] Short direction abbreviations")
|
|
|
|
|
|
def test_case_insensitivity():
|
|
"""Test that parsing is case-insensitive."""
|
|
assert parse_action("action: go south").type == ActionType.GO
|
|
assert parse_action("ACTION: GO SOUTH").type == ActionType.GO
|
|
assert parse_action("Action: Go South").type == ActionType.GO
|
|
assert parse_action("action: GO south").type == ActionType.GO
|
|
|
|
print(" [PASS] Case insensitivity")
|
|
|
|
|
|
def test_fallback_patterns():
|
|
"""Test fallback patterns without 'Action:' prefix."""
|
|
# Natural language variations
|
|
assert parse_action("I think I'll GO WEST to explore").type == ActionType.GO
|
|
assert parse_action("I'll GO NORTH").type == ActionType.GO
|
|
assert parse_action("Let me GO EAST").type == ActionType.GO
|
|
|
|
# Move variations
|
|
assert parse_action("I should move NORTH").type == ActionType.GO
|
|
assert parse_action("Let me head SOUTH").type == ActionType.GO
|
|
|
|
print(" [PASS] Fallback patterns")
|
|
|
|
|
|
def test_wait_action():
|
|
"""Test WAIT action parsing."""
|
|
assert parse_action("Action: WAIT").type == ActionType.WAIT
|
|
assert parse_action("I'll WAIT here").type == ActionType.WAIT
|
|
assert parse_action("Let me WAIT and see").type == ActionType.WAIT
|
|
|
|
print(" [PASS] WAIT action")
|
|
|
|
|
|
def test_look_action():
|
|
"""Test LOOK action parsing."""
|
|
assert parse_action("Action: LOOK").type == ActionType.LOOK
|
|
assert parse_action("Action: LOOK AT door").type == ActionType.LOOK
|
|
assert parse_action("Action: LOOK AT door").args == ("DOOR",)
|
|
|
|
print(" [PASS] LOOK action")
|
|
|
|
|
|
def test_invalid_actions():
|
|
"""Test that invalid actions are properly flagged."""
|
|
result = parse_action("I'm not sure what to do")
|
|
assert result.type == ActionType.INVALID
|
|
|
|
result = parse_action("Let me think about this...")
|
|
assert result.type == ActionType.INVALID
|
|
|
|
result = parse_action("The weather is nice today")
|
|
assert result.type == ActionType.INVALID
|
|
|
|
print(" [PASS] Invalid action detection")
|
|
|
|
|
|
def test_raw_match_capture():
|
|
"""Test that raw_match captures the matched text."""
|
|
result = parse_action("After thinking, Action: GO NORTH is best")
|
|
assert "GO NORTH" in result.raw_match
|
|
|
|
print(" [PASS] Raw match capture")
|
|
|
|
|
|
def test_embedded_actions():
|
|
"""Test extraction of actions embedded in longer text."""
|
|
long_response = """
|
|
Looking at the screenshot, I can see I'm in a dungeon corridor.
|
|
There's a rat to the east and a wall to the north.
|
|
The path south appears clear.
|
|
|
|
I think the best course of action is to investigate the rat.
|
|
|
|
Action: GO EAST
|
|
"""
|
|
|
|
result = parse_action(long_response)
|
|
assert result.type == ActionType.GO
|
|
assert result.args == ("EAST",)
|
|
|
|
print(" [PASS] Embedded action extraction")
|
|
|
|
|
|
def test_complex_actions():
|
|
"""Test more complex action types."""
|
|
# TAKE action
|
|
assert parse_action("Action: TAKE sword").type == ActionType.TAKE
|
|
assert parse_action("Action: TAKE sword").args == ("SWORD",)
|
|
|
|
# DROP action
|
|
assert parse_action("Action: DROP shield").type == ActionType.DROP
|
|
|
|
# USE action
|
|
assert parse_action("Action: USE key").type == ActionType.USE
|
|
assert parse_action("Action: USE key ON door").type == ActionType.USE
|
|
|
|
# OPEN/CLOSE
|
|
assert parse_action("Action: OPEN chest").type == ActionType.OPEN
|
|
assert parse_action("Action: CLOSE door").type == ActionType.CLOSE
|
|
|
|
print(" [PASS] Complex action types")
|
|
|
|
|
|
def test_push_action():
|
|
"""Test PUSH action with direction."""
|
|
result = parse_action("Action: PUSH boulder NORTH")
|
|
assert result.type == ActionType.PUSH
|
|
assert result.args == ("BOULDER", "NORTH")
|
|
|
|
result = parse_action("Action: PUSH box E")
|
|
assert result.type == ActionType.PUSH
|
|
assert result.args == ("BOX", "EAST")
|
|
|
|
print(" [PASS] PUSH action")
|
|
|
|
|
|
def test_speak_announce_actions():
|
|
"""Test SPEAK and ANNOUNCE with quoted strings."""
|
|
result = parse_action('Action: SPEAK "Hello there!"')
|
|
assert result.type == ActionType.SPEAK
|
|
assert result.args[0] == "HELLO THERE!" # Uppercase due to text normalization
|
|
|
|
result = parse_action("Action: ANNOUNCE 'Watch out!'")
|
|
assert result.type == ActionType.ANNOUNCE
|
|
|
|
print(" [PASS] SPEAK/ANNOUNCE actions")
|
|
|
|
|
|
def run_all_tests():
|
|
"""Run all parser tests."""
|
|
print("=" * 60)
|
|
print("Action Parser Tests")
|
|
print("=" * 60)
|
|
|
|
tests = [
|
|
test_explicit_go_directions,
|
|
test_short_directions,
|
|
test_case_insensitivity,
|
|
test_fallback_patterns,
|
|
test_wait_action,
|
|
test_look_action,
|
|
test_invalid_actions,
|
|
test_raw_match_capture,
|
|
test_embedded_actions,
|
|
test_complex_actions,
|
|
test_push_action,
|
|
test_speak_announce_actions,
|
|
]
|
|
|
|
passed = 0
|
|
failed = 0
|
|
|
|
for test in tests:
|
|
try:
|
|
test()
|
|
passed += 1
|
|
except AssertionError as e:
|
|
print(f" [FAIL] {test.__name__}: {e}")
|
|
failed += 1
|
|
except Exception as e:
|
|
print(f" [ERROR] {test.__name__}: {e}")
|
|
failed += 1
|
|
|
|
print("=" * 60)
|
|
print(f"Results: {passed} passed, {failed} failed")
|
|
print("=" * 60)
|
|
|
|
return failed == 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
success = run_all_tests()
|
|
sys.exit(0 if success else 1)
|