Add pattern tests

test/patterns/n_chase.py

@@ -0,0 +1,138 @@
+#!/usr/bin/env python3
+import uasyncio as asyncio
+import utime
+from machine import WDT
+from settings import Settings
+from patterns import Patterns
+
+
+async def main():
+    s = Settings()
+    pin = s.get("led_pin", 10)
+    num = s.get("num_leds", 30)
+
+    p = Patterns(pin=pin, num_leds=num)
+    wdt = WDT(timeout=10000)
+    
+    # Test 1: Basic n_chase (n1=5, n2=5, n3=1, n4=1)
+    print("Test 1: Basic n_chase (n1=5, n2=5, n3=1, n4=1)")
+    p.set_param("br", 255)
+    p.set_param("dl", 200)
+    p.set_param("n1", 5)  # 5 LEDs color0
+    p.set_param("n2", 5)  # 5 LEDs color1
+    p.set_param("n3", 1)  # Move 1 forward on even steps
+    p.set_param("n4", 1)  # Move 1 forward on odd steps
+    p.set_param("cl", [(255, 0, 0), (0, 255, 0)])  # Red and Green
+    p.select("n_chase")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 2: Forward and backward (n3=2, n4=-1)
+    print("Test 2: Forward and backward (n3=2, n4=-1)")
+    p.stopped = False
+    p.set_param("n1", 3)
+    p.set_param("n2", 3)
+    p.set_param("n3", 2)   # Move 2 forward on even steps
+    p.set_param("n4", -1)  # Move 1 backward on odd steps
+    p.set_param("dl", 150)
+    p.set_param("cl", [(0, 0, 255), (255, 255, 0)])  # Blue and Yellow
+    p.select("n_chase")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 3: Large segments (n1=10, n2=5)
+    print("Test 3: Large segments (n1=10, n2=5, n3=3, n4=3)")
+    p.stopped = False
+    p.set_param("n1", 10)  # 10 LEDs color0
+    p.set_param("n2", 5)   # 5 LEDs color1
+    p.set_param("n3", 3)   # Move 3 forward
+    p.set_param("n4", 3)   # Move 3 forward
+    p.set_param("dl", 200)
+    p.set_param("cl", [(255, 128, 0), (128, 0, 255)])  # Orange and Purple
+    p.select("n_chase")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 4: Fast movement (n3=5, n4=5)
+    print("Test 4: Fast movement (n3=5, n4=5)")
+    p.stopped = False
+    p.set_param("n1", 4)
+    p.set_param("n2", 4)
+    p.set_param("n3", 5)   # Move 5 forward
+    p.set_param("n4", 5)   # Move 5 forward
+    p.set_param("dl", 100)
+    p.set_param("cl", [(255, 0, 255), (0, 255, 255)])  # Magenta and Cyan
+    p.select("n_chase")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 2000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 5: Backward movement (n3=-2, n4=-2)
+    print("Test 5: Backward movement (n3=-2, n4=-2)")
+    p.stopped = False
+    p.set_param("n1", 6)
+    p.set_param("n2", 4)
+    p.set_param("n3", -2)  # Move 2 backward
+    p.set_param("n4", -2)  # Move 2 backward
+    p.set_param("dl", 200)
+    p.set_param("cl", [(255, 255, 255), (0, 0, 0)])  # White and Black
+    p.select("n_chase")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Test 6: Alternating forward/backward (n3=3, n4=-2)
+    print("Test 6: Alternating forward/backward (n3=3, n4=-2)")
+    p.stopped = False
+    p.set_param("n1", 5)
+    p.set_param("n2", 5)
+    p.set_param("n3", 3)   # Move 3 forward on even steps
+    p.set_param("n4", -2) # Move 2 backward on odd steps
+    p.set_param("dl", 250)
+    p.set_param("cl", [(255, 0, 0), (0, 255, 0)])  # Red and Green
+    p.select("n_chase")
+    task = asyncio.create_task(p.run())
+    start = utime.ticks_ms()
+    while utime.ticks_diff(utime.ticks_ms(), start) < 4000:
+        wdt.feed()
+        await asyncio.sleep_ms(10)
+    await p.stop()
+    await task
+    
+    # Cleanup
+    print("Test complete, turning off")
+    p.stopped = False
+    p.select("off")
+    task = asyncio.create_task(p.run())
+    await asyncio.sleep_ms(100)
+    await p.stop()
+    await task
+
+
+if __name__ == "__main__":
+    asyncio.run(main())
+
+