Rename the driver module and update imports so tests and main entry use the new presets naming, while moving Preset to its own file. Co-authored-by: Cursor <cursoragent@cursor.com>
82 lines
2.0 KiB
Python
82 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
import utime
|
|
from machine import WDT
|
|
from settings import Settings
|
|
from presets import Presets
|
|
|
|
|
|
def run_for(p, wdt, ms):
|
|
"""Helper: run current pattern for given ms using tick()."""
|
|
start = utime.ticks_ms()
|
|
while utime.ticks_diff(utime.ticks_ms(), start) < ms:
|
|
wdt.feed()
|
|
p.tick()
|
|
utime.sleep_ms(10)
|
|
|
|
|
|
def main():
|
|
s = Settings()
|
|
pin = s.get("led_pin", 10)
|
|
num = s.get("num_leds", 30)
|
|
|
|
p = Presets(pin=pin, num_leds=num)
|
|
wdt = WDT(timeout=10000)
|
|
|
|
# Test 1: Simple two-color transition
|
|
print("Test 1: Two-color transition (red <-> blue, delay=1000)")
|
|
p.edit("transition1", {
|
|
"p": "transition",
|
|
"b": 255,
|
|
"d": 1000, # transition duration
|
|
"c": [(255, 0, 0), (0, 0, 255)],
|
|
"a": True,
|
|
})
|
|
p.select("transition1")
|
|
run_for(p, wdt, 6000)
|
|
|
|
# Test 2: Multi-color transition
|
|
print("Test 2: Multi-color transition (red -> green -> blue -> white)")
|
|
p.edit("transition2", {
|
|
"p": "transition",
|
|
"d": 800,
|
|
"c": [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 255)],
|
|
"a": True,
|
|
})
|
|
p.select("transition2")
|
|
run_for(p, wdt, 8000)
|
|
|
|
# Test 3: One-shot transition (auto=False)
|
|
print("Test 3: One-shot transition (auto=False)")
|
|
p.edit("transition3", {
|
|
"p": "transition",
|
|
"d": 1000,
|
|
"c": [(255, 0, 0), (0, 255, 0)],
|
|
"a": False,
|
|
})
|
|
p.select("transition3")
|
|
# Run long enough for a single transition step
|
|
run_for(p, wdt, 2000)
|
|
|
|
# Test 4: Single-color behavior (should just stay on)
|
|
print("Test 4: Single-color transition (should hold color)")
|
|
p.edit("transition4", {
|
|
"p": "transition",
|
|
"c": [(0, 0, 255)],
|
|
"d": 500,
|
|
"a": True,
|
|
})
|
|
p.select("transition4")
|
|
run_for(p, wdt, 3000)
|
|
|
|
# Cleanup
|
|
print("Test complete, turning off")
|
|
p.edit("cleanup_off", {"p": "off"})
|
|
p.select("cleanup_off")
|
|
run_for(p, wdt, 200)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
|