jimmy
43957adb28
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>
152 lines
3.7 KiB
Python
152 lines
3.7 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: Basic rainbow with auto=True (continuous)
|
|
print("Test 1: Basic rainbow (auto=True, n1=1)")
|
|
p.edit("rainbow1", {
|
|
"p": "rainbow",
|
|
"b": 255,
|
|
"d": 100,
|
|
"n1": 1,
|
|
"a": True,
|
|
})
|
|
p.select("rainbow1")
|
|
run_for(p, wdt, 3000)
|
|
|
|
# Test 2: Fast rainbow
|
|
print("Test 2: Fast rainbow (low delay, n1=1)")
|
|
p.edit("rainbow2", {
|
|
"p": "rainbow",
|
|
"d": 50,
|
|
"n1": 1,
|
|
"a": True,
|
|
})
|
|
p.select("rainbow2")
|
|
run_for(p, wdt, 2000)
|
|
|
|
# Test 3: Slow rainbow
|
|
print("Test 3: Slow rainbow (high delay, n1=1)")
|
|
p.edit("rainbow3", {
|
|
"p": "rainbow",
|
|
"d": 500,
|
|
"n1": 1,
|
|
"a": True,
|
|
})
|
|
p.select("rainbow3")
|
|
run_for(p, wdt, 3000)
|
|
|
|
# Test 4: Low brightness rainbow
|
|
print("Test 4: Low brightness rainbow (n1=1)")
|
|
p.edit("rainbow4", {
|
|
"p": "rainbow",
|
|
"b": 64,
|
|
"d": 100,
|
|
"n1": 1,
|
|
"a": True,
|
|
})
|
|
p.select("rainbow4")
|
|
run_for(p, wdt, 2000)
|
|
|
|
# Test 5: Single-step rainbow (auto=False)
|
|
print("Test 5: Single-step rainbow (auto=False, n1=1)")
|
|
p.edit("rainbow5", {
|
|
"p": "rainbow",
|
|
"b": 255,
|
|
"d": 100,
|
|
"n1": 1,
|
|
"a": False,
|
|
})
|
|
p.step = 0
|
|
for i in range(10):
|
|
p.select("rainbow5")
|
|
# One tick advances the generator one frame when auto=False
|
|
p.tick()
|
|
utime.sleep_ms(100)
|
|
wdt.feed()
|
|
|
|
# Test 6: Verify step updates correctly
|
|
print("Test 6: Verify step updates (auto=False, n1=1)")
|
|
p.edit("rainbow6", {
|
|
"p": "rainbow",
|
|
"n1": 1,
|
|
"a": False,
|
|
})
|
|
initial_step = p.step
|
|
p.select("rainbow6")
|
|
p.tick()
|
|
final_step = p.step
|
|
print(f"Step updated from {initial_step} to {final_step} (expected increment: 1)")
|
|
|
|
# Test 7: Fast step increment (n1=5)
|
|
print("Test 7: Fast rainbow (n1=5, auto=True)")
|
|
p.edit("rainbow7", {
|
|
"p": "rainbow",
|
|
"b": 255,
|
|
"d": 100,
|
|
"n1": 5,
|
|
"a": True,
|
|
})
|
|
p.select("rainbow7")
|
|
run_for(p, wdt, 2000)
|
|
|
|
# Test 8: Very fast step increment (n1=10)
|
|
print("Test 8: Very fast rainbow (n1=10, auto=True)")
|
|
p.edit("rainbow8", {
|
|
"p": "rainbow",
|
|
"n1": 10,
|
|
"a": True,
|
|
})
|
|
p.select("rainbow8")
|
|
run_for(p, wdt, 2000)
|
|
|
|
# Test 9: Verify n1 controls step increment (auto=False)
|
|
print("Test 9: Verify n1 step increment (auto=False, n1=5)")
|
|
p.edit("rainbow9", {
|
|
"p": "rainbow",
|
|
"n1": 5,
|
|
"a": False,
|
|
})
|
|
p.step = 0
|
|
initial_step = p.step
|
|
p.select("rainbow9")
|
|
p.tick()
|
|
final_step = p.step
|
|
expected_step = (initial_step + 5) % 256
|
|
print(f"Step updated from {initial_step} to {final_step} (expected: {expected_step})")
|
|
if final_step == expected_step:
|
|
print("✓ n1 step increment working correctly")
|
|
else:
|
|
print(f"✗ Step increment mismatch! Expected {expected_step}, got {final_step}")
|
|
|
|
# Cleanup
|
|
print("Test complete, turning off")
|
|
p.edit("cleanup_off", {"p": "off"})
|
|
p.select("cleanup_off")
|
|
run_for(p, wdt, 100)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|