Initial commit

tests/test_mounting.py

@@ -0,0 +1,153 @@
+"""Mounting traps on other traps."""
+
+from conftest import step, DT
+from game.traps import Block, Spike
+
+
+PLATFORM_WITH_SPIKE = """
+name: t
+tile_size: 32
+map: |
+  ############
+  #..........#
+  #..........#
+  #.........G#
+  #P.........#
+  ############
+traps:
+  - type: block
+    at: [3, 2]
+    path: [[3, 2], [8, 2]]
+    mode: pingpong
+    speed: 120
+    sprite: patrol_block
+    mounts:
+      - type: spike
+        at: [0, -1]
+        trigger: always
+        direction: up
+"""
+
+
+def test_spike_mount_follows_platform(make_game):
+    g = make_game(PLATFORM_WITH_SPIKE)
+    plat = g.level.traps[0]
+    sp = plat.mounts[0]
+    assert isinstance(sp, Spike)
+    for _ in range(30):
+        g.level.update(DT, g)
+    p = plat.current_rect()
+    hz = sp.hazard_rects()[0]
+    # spike sits one tile above the platform and moves with it
+    assert hz.centerx == p.centerx
+    assert hz.bottom <= p.top + 1
+
+
+def test_block_mounted_on_block(make_game):
+    g = make_game("""
+name: t
+tile_size: 32
+map: |
+  ############
+  #..........#
+  #..........#
+  #.........G#
+  #P.........#
+  ############
+traps:
+  - type: block
+    at: [3, 2]
+    path: [[3, 2], [8, 2]]
+    mode: pingpong
+    speed: 120
+    sprite: patrol_block
+    mounts:
+      - type: block
+        at: [-1, 0]
+        deadly: true
+      - type: block
+        at: [1, 0]
+        deadly: true
+""")
+    plat = g.level.traps[0]
+    g.level.update(DT, g)
+    for m in plat.mounts:
+        assert m._mounted and m.deadly
+        # each deadly mount tracks the platform at its offset and is lethal
+        exp = (plat.current_rect().x + m._mount_off[0],
+               plat.current_rect().y + m._mount_off[1])
+        assert (m.current_rect().x, m.current_rect().y) == exp
+        assert m.hazard_rects()
+
+
+MOBILE_MOUNT = """
+name: t
+tile_size: 32
+map: |
+  ################
+  #..............#
+  #..............#
+  #..............#
+  #..............#
+  #P............G#
+  ################
+traps:
+  - type: block
+    at: [3, 4]
+    path: [[3, 4], [8, 4]]
+    mode: pingpong
+    speed: 120
+    sprite: patrol_block
+    mounts:
+      # Rides the patrol AND runs its own upward stroke when armed.
+      - type: block
+        at: [0, -1]
+        move: [0, -3]
+        speed: 240
+        mode: once
+        trigger: { within: 40 }   # player is always in range -> stays extended
+        mounts:
+          - type: spike
+            at: [0, -1]
+            trigger: always
+            direction: up
+"""
+
+
+def test_block_mount_runs_own_motion_while_riding(make_game):
+    g = make_game(MOBILE_MOUNT)
+    plat = g.level.traps[0]
+    mnt = plat.mounts[0]
+    assert isinstance(mnt, Block) and mnt._mounted
+    tile = g.level.tile
+
+    # At rest the mount sits one tile above the patrol, tracking its column.
+    assert mnt.current_rect().x == plat.current_rect().x
+    assert mnt.current_rect().top == plat.current_rect().top - tile
+
+    for _ in range(120):
+        g.level.update(DT, g)
+
+    pr, mr = plat.current_rect(), mnt.current_rect()
+    # Still rides along horizontally (same column offset as home)...
+    assert mr.x == pr.x
+    # ...but has run its own three-tile upward stroke above the resting spot
+    # (resting = 1 tile up; extended = 1 + 3 tiles up).
+    assert mr.top == pr.top - 4 * tile
+    # It reports as a solid carrier so a rider would be carried.
+    assert mnt.solid_rects() and mnt.carriers()
+    # The spike rides the lunging block, one tile above it.
+    spike = mnt.mounts[0]
+    hz = spike.hazard_rects()[0]
+    assert hz.centerx == mr.centerx and hz.bottom <= mr.top + 1
+
+
+def test_reset_repositions_mounts(make_game):
+    g = make_game(PLATFORM_WITH_SPIKE)
+    plat = g.level.traps[0]
+    sp = plat.mounts[0]
+    for _ in range(40):
+        g.level.update(DT, g)
+    g.level.reset()
+    # after reset the platform is home and the mount snapped back onto it
+    assert sp.base_rect.centerx == plat.current_rect().centerx