125 lines
3.8 KiB
Python
125 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test mcrfpy.step() function (#153)
|
|
===================================
|
|
|
|
Tests the Python-controlled simulation advancement for headless mode.
|
|
|
|
Key behavior:
|
|
- step(dt) advances simulation by dt seconds
|
|
- step(None) or step() advances to next scheduled event
|
|
- Returns actual time advanced
|
|
- In windowed mode, returns 0.0 (no-op)
|
|
"""
|
|
|
|
import mcrfpy
|
|
import sys
|
|
|
|
def run_tests():
|
|
"""Run step() function tests"""
|
|
print("=== mcrfpy.step() Tests ===\n")
|
|
|
|
# Test 1: step() with specific dt value
|
|
print("Test 1: step() with specific dt value")
|
|
dt = mcrfpy.step(0.1) # Advance 100ms
|
|
print(f" step(0.1) returned: {dt}")
|
|
# In headless mode, should return 0.1
|
|
# In windowed mode, returns 0.0
|
|
if dt == 0.0:
|
|
print(" Note: Running in windowed mode - step() is no-op")
|
|
else:
|
|
assert abs(dt - 0.1) < 0.001, f"Expected ~0.1, got {dt}"
|
|
print(" Correctly advanced by 0.1 seconds")
|
|
print()
|
|
|
|
# Test 2: step() with integer value (converts to float)
|
|
print("Test 2: step() with integer value")
|
|
dt = mcrfpy.step(1) # Advance 1 second
|
|
print(f" step(1) returned: {dt}")
|
|
if dt != 0.0:
|
|
assert abs(dt - 1.0) < 0.001, f"Expected ~1.0, got {dt}"
|
|
print(" Correctly advanced by 1.0 seconds")
|
|
print()
|
|
|
|
# Test 3: step(None) - advance to next event
|
|
print("Test 3: step(None) - advance to next event")
|
|
dt = mcrfpy.step(None)
|
|
print(f" step(None) returned: {dt}")
|
|
if dt != 0.0:
|
|
assert dt >= 0, "step(None) should return non-negative dt"
|
|
print(f" Advanced by {dt} seconds to next event")
|
|
print()
|
|
|
|
# Test 4: step() with no argument (same as step(None))
|
|
print("Test 4: step() with no argument")
|
|
dt = mcrfpy.step()
|
|
print(f" step() returned: {dt}")
|
|
if dt != 0.0:
|
|
assert dt >= 0, "step() should return non-negative dt"
|
|
print(f" Advanced by {dt} seconds")
|
|
print()
|
|
|
|
# Test 5: Timer callback with step()
|
|
print("Test 5: Timer fires after step() advances past interval")
|
|
timer_fired = [False] # Use list for mutable closure
|
|
|
|
def on_timer(runtime):
|
|
"""Timer callback - receives runtime in ms"""
|
|
timer_fired[0] = True
|
|
print(f" Timer fired at simulation time={runtime}ms")
|
|
|
|
# Set a timer for 500ms
|
|
mcrfpy.setTimer("test_timer", on_timer, 500)
|
|
|
|
# Step 600ms - timer should fire (500ms interval + some buffer)
|
|
dt = mcrfpy.step(0.6)
|
|
if dt != 0.0: # Headless mode
|
|
# Timer should have fired
|
|
if timer_fired[0]:
|
|
print(" Timer correctly fired after step(0.6)")
|
|
else:
|
|
# Try another step to ensure timer fires
|
|
mcrfpy.step(0.1)
|
|
if timer_fired[0]:
|
|
print(" Timer fired after additional step")
|
|
else:
|
|
print(" WARNING: Timer didn't fire - check timer synchronization")
|
|
else:
|
|
print(" Skipping timer test in windowed mode")
|
|
|
|
# Clean up
|
|
mcrfpy.delTimer("test_timer")
|
|
print()
|
|
|
|
# Test 6: Error handling - invalid argument type
|
|
print("Test 6: Error handling - invalid argument type")
|
|
try:
|
|
mcrfpy.step("invalid")
|
|
print(" ERROR: Should have raised TypeError")
|
|
return False
|
|
except TypeError as e:
|
|
print(f" Correctly raised TypeError: {e}")
|
|
print()
|
|
|
|
print("=== All step() Tests Passed! ===")
|
|
return True
|
|
|
|
# Main execution
|
|
if __name__ == "__main__":
|
|
try:
|
|
# Create a scene for the test
|
|
mcrfpy.createScene("test_step")
|
|
mcrfpy.setScene("test_step")
|
|
|
|
if run_tests():
|
|
print("\nPASS")
|
|
sys.exit(0)
|
|
else:
|
|
print("\nFAIL")
|
|
sys.exit(1)
|
|
except Exception as e:
|
|
print(f"\nFAIL: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
sys.exit(1)
|