92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
|
"""Trigger conditions and the arm-delay hysteresis."""
|
||
|
|
|
||
|
|
import pygame
|
||
|
|
from conftest import FakeGame, FakeLevel
|
||
|
|
from game.traps import make_condition
|
||
|
|
|
||
|
|
|
||
|
|
def ev(spec, px, py, w=20, h=28, dt=0.0):
|
||
|
|
trap = type("T", (), {"tile": 32,
|
||
|
|
"sensor_rect": lambda self: pygame.Rect(100, 100, 32, 32)})()
|
||
|
|
return make_condition(spec).evaluate(trap, FakeGame(pygame.Rect(px, py, w, h)), dt)
|
||
|
|
# trap centre = (116, 116); left100 right132 top100 bottom132
|
||
|
|
|
||
|
|
|
||
|
|
def test_always():
|
||
|
|
assert ev("always", 999, 999) is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_within_radius():
|
||
|
|
assert ev({"within": 2}, 106, 104) is True # ~10px away
|
||
|
|
assert ev({"within": 2}, 400, 116) is False # far
|
||
|
|
|
||
|
|
|
||
|
|
def test_dir_left_right():
|
||
|
|
assert ev({"dir": "left"}, 40, 105) is True # to the left
|
||
|
|
assert ev({"dir": "left"}, 200, 105) is False # to the right
|
||
|
|
assert ev({"dir": "right"}, 200, 105) is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_dir_range():
|
||
|
|
assert ev({"dir": "left", "range": 2}, 60, 105) is True # within 2 tiles left
|
||
|
|
assert ev({"dir": "left", "range": 2}, 10, 105) is False # too far left
|
||
|
|
|
||
|
|
|
||
|
|
def test_dir_aligned():
|
||
|
|
# above + aligned requires horizontal overlap with the trap column
|
||
|
|
assert ev({"dir": "above", "aligned": True}, 108, 60) is True
|
||
|
|
assert ev({"dir": "above", "aligned": True}, 108, 110) is False # not above
|
||
|
|
assert ev({"dir": "above", "aligned": True}, 400, 60) is False # not aligned
|
||
|
|
# without aligned, any column counts
|
||
|
|
assert ev({"dir": "above"}, 400, 60) is True
|
||
|
|
|
||
|
|
|
||
|
|
def test_all_and_any():
|
||
|
|
c = {"all": [{"within": 3}, {"dir": "above", "aligned": True}]}
|
||
|
|
assert ev(c, 108, 80) is True
|
||
|
|
assert ev(c, 108, 110) is False # close but not above
|
||
|
|
c2 = {"any": [{"dir": "left"}, {"dir": "right"}]}
|
||
|
|
assert ev(c2, 40, 105) is True
|
||
|
|
assert ev(c2, 110, 40) is False # above-only satisfies neither
|
||
|
|
|
||
|
|
|
||
|
|
def test_timer_cycles():
|
||
|
|
c = make_condition({"timer": {"interval": 0.3, "up_time": 0.2}})
|
||
|
|
trap = type("T", (), {"tile": 32,
|
||
|
|
"sensor_rect": lambda self: pygame.Rect(0, 0, 32, 32)})()
|
||
|
|
g = FakeGame(pygame.Rect(0, 0, 4, 4))
|
||
|
|
states = [c.evaluate(trap, g, 1 / 60) for _ in range(30)]
|
||
|
|
assert states[0] is False # starts in the "off" interval
|
||
|
|
assert any(states) and not all(states) # cycles on and off
|
||
|
|
|
||
|
|
|
||
|
|
def test_bad_condition_raises():
|
||
|
|
import pytest
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
make_condition({"nope": 1})
|
||
|
|
|
||
|
|
|
||
|
|
def test_delay_hysteresis(make_level):
|
||
|
|
# A spike within-1.6 with delay 0.25 arms only after staying in range.
|
||
|
|
lvl = make_level("""
|
||
|
|
name: t
|
||
|
|
tile_size: 32
|
||
|
|
map: |
|
||
|
|
#####
|
||
|
|
#P.G#
|
||
|
|
#####
|
||
|
|
traps:
|
||
|
|
- type: spike
|
||
|
|
at: [1, 1]
|
||
|
|
trigger: { within: 5 }
|
||
|
|
delay: 0.25
|
||
|
|
""")
|
||
|
|
sp = lvl.traps[0]
|
||
|
|
g = FakeGame(sp.base_rect.copy()) # player right on it -> in range
|
||
|
|
armed = None
|
||
|
|
for i in range(30):
|
||
|
|
if sp.triggered(g, 1 / 60):
|
||
|
|
armed = i
|
||
|
|
break
|
||
|
|
assert armed is not None and 13 <= armed <= 17 # ~0.25s = 15 frames
|