60 lines
2.5 KiB
Python
60 lines
2.5 KiB
Python
import asyncio
|
|
import pytest
|
|
from unittest.mock import patch, MagicMock, AsyncMock
|
|
|
|
from src.sim.simulator import Simulator
|
|
from src.classes.action.move_to_direction import MoveToDirection
|
|
from src.classes.tile import TileType
|
|
from src.classes.action_runtime import ActionInstance
|
|
|
|
def test_simulator_step_moves_avatar_and_sets_tile(base_world, dummy_avatar):
|
|
# Set initial position
|
|
dummy_avatar.pos_x = 1
|
|
dummy_avatar.pos_y = 1
|
|
# Ensure tile is updated to initial position (fixture puts it at 0,0)
|
|
dummy_avatar.tile = base_world.map.get_tile(1, 1)
|
|
|
|
sim = Simulator(base_world)
|
|
base_world.avatar_manager.avatars[dummy_avatar.id] = dummy_avatar
|
|
|
|
# Manually assign a MoveToDirection action to avoid relying on LLM
|
|
action = MoveToDirection(dummy_avatar, base_world)
|
|
# "East" means x + 1
|
|
direction = "East"
|
|
action.start(direction=direction) # Initialize start_monthstamp etc.
|
|
|
|
# Wrap in ActionInstance
|
|
dummy_avatar.current_action = ActionInstance(action=action, params={"direction": direction})
|
|
|
|
# Mock LLM to avoid external calls or errors
|
|
# We need to patch where it is imported or used.
|
|
# simulator.py imports process_avatar_long_term_objective, etc.
|
|
# The easiest way is to patch call_llm globally or specifically the phases.
|
|
with patch("src.sim.simulator.llm_ai") as mock_ai, \
|
|
patch("src.sim.simulator.process_avatar_long_term_objective", new_callable=AsyncMock) as mock_lto, \
|
|
patch("src.classes.nickname.process_avatar_nickname", new_callable=AsyncMock) as mock_nick, \
|
|
patch("src.classes.relation_resolver.RelationResolver.run_batch", new_callable=AsyncMock) as mock_rr:
|
|
|
|
mock_ai.decide = MagicMock(return_value={})
|
|
# Mock async returns for gathered tasks
|
|
mock_lto.return_value = None
|
|
mock_nick.return_value = None
|
|
mock_rr.return_value = []
|
|
|
|
print(f"DEBUG: Before step: pos_x={dummy_avatar.pos_x}")
|
|
# Run step synchronously
|
|
asyncio.run(sim.step())
|
|
print(f"DEBUG: After step: pos_x={dummy_avatar.pos_x}")
|
|
print(f"DEBUG: move_step_length={getattr(dummy_avatar, 'move_step_length', 'Not set')}")
|
|
print(f"DEBUG: effects={dummy_avatar.effects}")
|
|
|
|
# Assert moved East (x increased by move_step_length)
|
|
# Current move step for Qi Refinement is 2
|
|
assert dummy_avatar.pos_x == 3
|
|
assert dummy_avatar.pos_y == 1
|
|
|
|
# Assert tile is updated
|
|
assert dummy_avatar.tile is not None
|
|
assert dummy_avatar.tile.x == 3
|
|
assert dummy_avatar.tile.y == 1
|