test: physics and player movement coverage

tests/test_player.py

@@ -0,0 +1,165 @@
+import pytest
+
+from game import constants
+from tests.support import makeWorld, settle
+from game.sim.inputCommand import InputCommand
+
+
+ledgeLevel = [
+    "        ",
+    "        ",
+    " p      ",
+    "XXX     ",
+]
+
+flatLevel = [
+    " p      ",
+    "        ",
+    "XXXXXXXX",
+]
+
+wallLevel = [
+    "    #",
+    "    #",
+    "    #",
+    "    #",
+    "XXXXX",
+]
+
+
+def walkOffLedge(world):
+    for _ in range(40):
+        world.step(InputCommand(right=True, run=True))
+        if not world.player.onGround:
+            return True
+    return False
+
+
+def test_coyoteJumpKeepsAirJump():
+    world = makeWorld(ledgeLevel)
+    assert walkOffLedge(world)
+    world.step(InputCommand(right=True, jump=True, jumpHeld=True))
+    player = world.player
+    assert player.vy == pytest.approx(constants.jumpVelocity + constants.gravity)
+    assert player.airJumps == constants.maxAirJumps
+
+
+def test_afterCoyoteExpiresJumpIsDoubleJump():
+    world = makeWorld(ledgeLevel)
+    assert walkOffLedge(world)
+    for _ in range(constants.coyoteFrames + 2):
+        world.step(InputCommand())
+    world.step(InputCommand(jump=True, jumpHeld=True))
+    player = world.player
+    assert player.airJumps == 0
+    assert player.vy == pytest.approx(constants.doubleJumpVelocity + constants.gravity)
+
+
+def test_doubleJumpConsumesAirJump():
+    world = makeWorld(flatLevel)
+    settle(world, 20)
+    world.step(InputCommand(jump=True, jumpHeld=True))
+    assert world.player.vy < 0
+    assert world.player.airJumps == constants.maxAirJumps
+    for _ in range(3):
+        world.step(InputCommand(jumpHeld=True))
+    world.step(InputCommand(jump=True, jumpHeld=True))
+    assert world.player.airJumps == 0
+    assert world.player.vy == pytest.approx(constants.doubleJumpVelocity + constants.gravity)
+
+
+def test_jumpBufferFiresOnLanding():
+    world = makeWorld(flatLevel)
+    player = world.player
+    player.airJumps = 0
+    player.coyoteTimer = 0
+    player.onGround = False
+    player.y = 2 * constants.tileSize - player.h - 5.0
+    world.step(InputCommand(jump=True))
+    minVy = 0.0
+    grounded = False
+    for _ in range(14):
+        world.step(InputCommand())
+        grounded = grounded or world.player.onGround
+        minVy = min(minVy, world.player.vy)
+    assert grounded
+    assert minVy == pytest.approx(constants.jumpVelocity + constants.gravity)
+
+
+def test_noBufferNoRejump():
+    world = makeWorld(flatLevel)
+    player = world.player
+    player.airJumps = 0
+    player.coyoteTimer = 0
+    player.onGround = False
+    player.y = 2 * constants.tileSize - player.h - 5.0
+    minVy = 0.0
+    for _ in range(14):
+        world.step(InputCommand())
+        minVy = min(minVy, world.player.vy)
+    assert minVy >= 0.0
+
+
+def test_dashLaunchesAndCoolsDown():
+    world = makeWorld(flatLevel)
+    settle(world, 20)
+    world.step(InputCommand(right=True, dash=True))
+    player = world.player
+    assert player.vx == pytest.approx(constants.dashSpeed)
+    assert player.dashTimer == constants.dashFrames - 1
+    assert player.vy == 0.0
+    assert player.dashCooldown > 0
+
+
+def test_wallSlideClampsFallSpeed():
+    world = makeWorld(wallLevel)
+    player = world.player
+    player.x = 4 * constants.tileSize - player.w
+    player.y = constants.tileSize
+    player.vy = 5.0
+    player.onGround = False
+    world.step(InputCommand(right=True))
+    assert player.wallSliding
+    assert player.vy == pytest.approx(constants.wallSlideSpeed)
+
+
+def test_noWallSlideWhenNotPressingIntoWall():
+    world = makeWorld(wallLevel)
+    player = world.player
+    player.x = 4 * constants.tileSize - player.w
+    player.y = constants.tileSize
+    player.vy = 5.0
+    player.onGround = False
+    world.step(InputCommand(left=True))
+    assert not player.wallSliding
+    assert player.vy > constants.wallSlideSpeed
+
+
+def test_wallJumpPushesAway():
+    world = makeWorld(wallLevel)
+    player = world.player
+    player.x = 4 * constants.tileSize - player.w
+    player.y = constants.tileSize
+    player.vy = 2.0
+    player.onGround = False
+    world.step(InputCommand(right=True, jump=True, jumpHeld=True))
+    assert player.vx == pytest.approx(-constants.wallJumpVelocityX)
+    assert player.vy == pytest.approx(constants.wallJumpVelocityY + constants.gravity)
+
+
+def test_variableJumpCutReducesRise():
+    holdWorld = makeWorld(flatLevel)
+    settle(holdWorld, 20)
+    holdWorld.step(InputCommand(jump=True, jumpHeld=True))
+    holdWorld.step(InputCommand(jumpHeld=True))
+    holdVy = holdWorld.player.vy
+
+    cutWorld = makeWorld(flatLevel)
+    settle(cutWorld, 20)
+    cutWorld.step(InputCommand(jump=True, jumpHeld=True))
+    cutWorld.step(InputCommand(jumpHeld=False))
+    cutVy = cutWorld.player.vy
+
+    assert holdVy < -6.0
+    assert cutVy > -4.0
+    assert cutVy > holdVy