File size: 862 Bytes
c50d37a
bd0ae14
c50d37a
 
 
 
bd0ae14
c50d37a
bd0ae14
c50d37a
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
# tests/web/test_multiagent_memory_route.py
"""Creating a predator_chase session returns multi-agent memory frames over HTTP."""
from proteus.web.local.server import handle_request
import proteus.game.scenarios  # noqa: F401


def test_predator_chase_session_returns_painted_multiagent_memory():
    registry: dict = {}
    body = {"scenario": "predator_chase", "difficulty": "easy", "seed": 7,
            "play_turns": 5, "memory": "default"}
    status, payload, _ = handle_request("POST", "/session", body, registry)
    assert status == 200, payload
    frames = payload["memory"]["frames"]
    assert frames, "expected memory frames"
    # Distractor blue (9) appears somewhere in an early frame's grid.
    flat0 = [v for row in frames[0]["grid"] for v in row]
    assert 9 in flat0
    # events key is present on frames.
    assert "events" in frames[0]