led-bar/test/patterns/transition.py

#!/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 transition with 2 colors (auto=True, cycles continuously)
    print("Test 1: Basic transition (2 colors, 1000ms delay, auto=True)")
    p.set_param("br", 255)
    p.set_param("dl", 1000)  # 1 second transition time
    p.set_param("auto", True)  # Cycle continuously
    p.set_param("cl", [(255, 0, 0), (0, 255, 0)])  # Red to Green
    p.select("transition")
    task = asyncio.create_task(p.run())
    start = utime.ticks_ms()
    # Run for 5 seconds to see multiple transitions
    while utime.ticks_diff(utime.ticks_ms(), start) < 5000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    p.stopped = True
    await task
    
    # Test 2: Fast transition (auto=True, cycles continuously)
    print("Test 2: Fast transition (500ms delay, auto=True)")
    p.stopped = False
    p.set_param("dl", 500)  # 500ms transition time
    p.set_param("auto", True)  # Cycle continuously
    p.set_param("cl", [(0, 0, 255), (255, 255, 0)])  # Blue to Yellow
    p.select("transition")
    task = asyncio.create_task(p.run())
    start = utime.ticks_ms()
    # Run for 3 seconds
    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    p.stopped = True
    await task
    
    # Test 3: Multiple colors transition (auto=True, cycles continuously)
    print("Test 3: Multiple colors transition (3 colors, auto=True)")
    p.stopped = False
    p.set_param("dl", 800)
    p.set_param("auto", True)  # Cycle continuously
    p.set_param("cl", [
        (255, 0, 0),   # Red
        (0, 255, 0),   # Green
        (0, 0, 255),   # Blue
    ])
    p.select("transition")
    task = asyncio.create_task(p.run())
    start = utime.ticks_ms()
    # Run for 8 seconds to see full cycles
    while utime.ticks_diff(utime.ticks_ms(), start) < 8000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    p.stopped = True
    await task
    
    # Test 4: Single color (should just stay that color)
    print("Test 4: Single color (should stay that color)")
    p.stopped = False
    p.set_param("dl", 1000)
    p.set_param("cl", [(255, 128, 0)])  # Orange
    p.select("transition")
    task = asyncio.create_task(p.run())
    start = utime.ticks_ms()
    # Run for 3 seconds
    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    p.stopped = True
    await task
    
    # Test 5: Many colors transition (auto=True, cycles continuously)
    print("Test 5: Many colors transition (5 colors, auto=True)")
    p.stopped = False
    p.set_param("dl", 600)
    p.set_param("auto", True)  # Cycle continuously
    p.set_param("cl", [
        (255, 0, 0),     # Red
        (255, 128, 0),   # Orange
        (255, 255, 0),   # Yellow
        (0, 255, 0),     # Green
        (0, 0, 255),     # Blue
    ])
    p.select("transition")
    task = asyncio.create_task(p.run())
    start = utime.ticks_ms()
    # Run for 10 seconds to see multiple cycles
    while utime.ticks_diff(utime.ticks_ms(), start) < 10000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    p.stopped = True
    await task
    
    # Test 6: Low brightness transition (auto=True, cycles continuously)
    print("Test 6: Low brightness transition (auto=True)")
    p.stopped = False
    p.set_param("br", 64)  # Low brightness
    p.set_param("dl", 1000)
    p.set_param("auto", True)  # Cycle continuously
    p.set_param("cl", [(255, 0, 0), (0, 255, 0)])
    p.select("transition")
    task = asyncio.create_task(p.run())
    start = utime.ticks_ms()
    # Run for 3 seconds
    while utime.ticks_diff(utime.ticks_ms(), start) < 3000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    p.stopped = True
    await task
    
    # Test 7: Single-shot transition (auto=False, only color1 to color2)
    print("Test 7: Single-shot transition (auto=False, color1 to color2 only)")
    p.stopped = False
    p.set_param("br", 255)
    p.set_param("dl", 1000)  # 1 second transition
    p.set_param("auto", False)  # Run only once
    p.set_param("cl", [
        (255, 0, 0),     # Red (color1)
        (0, 255, 0),     # Green (color2)
        (0, 0, 255),     # Blue (should be ignored)
        (255, 255, 0),   # Yellow (should be ignored)
    ])
    p.select("transition")
    task = asyncio.create_task(p.run())
    
    # The transition should complete once (color1 to color2) and then stop
    # Total time should be ~1000ms
    # Wait a bit longer to verify it doesn't continue
    start = utime.ticks_ms()
    while utime.ticks_diff(utime.ticks_ms(), start) < 2000:
        wdt.feed()
        await asyncio.sleep_ms(10)
    
    # Task should have completed on its own (not stopped manually)
    # Verify it's stopped
    if not p.stopped:
        print("Warning: Transition should have stopped automatically with auto=False")
        p.stopped = True
    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)
    p.stopped = True
    await task


if __name__ == "__main__":
    asyncio.run(main())