118 lines
3.6 KiB
Python
118 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script for rainbow pattern
|
|
Run with: mpremote run test/rainbow.py
|
|
"""
|
|
|
|
import patterns
|
|
import utime
|
|
from settings import Settings
|
|
from machine import WDT
|
|
|
|
def run_test_case(p, test_name, duration_ms=5000):
|
|
"""Run a test case for specified duration"""
|
|
print(f"\n--- {test_name} ---")
|
|
print(f"Parameters: n1={p.n1} nodes, delay={p.delay}ms cycle time, brightness={p.brightness}")
|
|
print(f"Running for {duration_ms//1000} seconds...")
|
|
|
|
# Initialize watchdog timer
|
|
wdt = WDT(timeout=10000)
|
|
wdt.feed()
|
|
|
|
# Start pattern
|
|
p.select("rb")
|
|
|
|
# Run for specified duration
|
|
start_time = utime.ticks_ms()
|
|
while utime.ticks_diff(utime.ticks_ms(), start_time) < duration_ms:
|
|
wdt.feed()
|
|
utime.sleep_ms(100)
|
|
|
|
# Stop pattern
|
|
p.run = False
|
|
print(f"{test_name} completed")
|
|
|
|
# Brief pause between tests
|
|
utime.sleep_ms(500)
|
|
|
|
def test_rainbow():
|
|
print("Testing rainbow pattern with multiple configurations...")
|
|
|
|
# Load settings
|
|
settings = Settings()
|
|
|
|
# Initialize patterns using settings
|
|
p = patterns.Patterns(
|
|
pin=settings["led_pin"],
|
|
num_leds=settings["num_leds"],
|
|
brightness=255, # Full brightness
|
|
delay=1000 # Base cycle time
|
|
)
|
|
|
|
print(f"LED Pin: {settings['led_pin']}")
|
|
print(f"LEDs: {settings['num_leds']}")
|
|
print(f"Base Brightness: {p.brightness}")
|
|
|
|
# Test Case 1: Single node rainbow (full spectrum)
|
|
p.n1 = 1 # 1 node (full spectrum)
|
|
p.delay = 3000 # 3 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 1: Single node (full spectrum)", 12000)
|
|
|
|
# Test Case 2: Two node rainbow (red/green)
|
|
p.n1 = 2 # 2 nodes
|
|
p.delay = 2000 # 2 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 2: Two nodes (red/green)", 10000)
|
|
|
|
# Test Case 3: Three node rainbow (RGB)
|
|
p.n1 = 3 # 3 nodes
|
|
p.delay = 1500 # 1.5 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 3: Three nodes (RGB)", 9000)
|
|
|
|
# Test Case 4: Four node rainbow (RYGB)
|
|
p.n1 = 4 # 4 nodes
|
|
p.delay = 1000 # 1 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 4: Four nodes (RYGB)", 8000)
|
|
|
|
# Test Case 5: Six node rainbow (ROYGBP)
|
|
p.n1 = 6 # 6 nodes
|
|
p.delay = 800 # 0.8 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 5: Six nodes (ROYGBP)", 8000)
|
|
|
|
# Test Case 6: Eight node rainbow
|
|
p.n1 = 8 # 8 nodes
|
|
p.delay = 600 # 0.6 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 6: Eight nodes", 6000)
|
|
|
|
# Test Case 7: Twelve node rainbow (fine gradation)
|
|
p.n1 = 12 # 12 nodes
|
|
p.delay = 400 # 0.4 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 7: Twelve nodes (fine gradation)", 6000)
|
|
|
|
# Test Case 8: Many nodes, slow cycle
|
|
p.n1 = 20 # 20 nodes
|
|
p.delay = 2000 # 2 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 8: Many nodes, slow cycle", 10000)
|
|
|
|
# Test Case 9: One second cycle
|
|
p.n1 = 5 # 5 nodes
|
|
p.delay = 1000 # 1 second cycle
|
|
p.brightness = 255 # Full brightness
|
|
run_test_case(p, "Test 9: One second cycle (5 nodes)", 6000)
|
|
|
|
print("\n=== All tests completed ===")
|
|
|
|
# Turn off LEDs
|
|
p.off()
|
|
print("LEDs turned off")
|
|
|
|
if __name__ == "__main__":
|
|
test_rainbow()
|