from lager import Net, NetType
import time
def setup_test():
"""Power on all USB peripherals for testing."""
usb_devices = ['PROGRAMMER', 'SENSOR', 'DEBUGGER']
for name in usb_devices:
try:
usb = Net.get(name, type=NetType.Usb)
usb.enable()
print(f"{name} powered on")
except Exception as e:
print(f"Warning: {name} - {e}")
time.sleep(1) # Wait for USB enumeration
# Enable main power
psu = Net.get('VDD', type=NetType.PowerSupply)
psu.set_voltage(3.3)
psu.enable()
return True
def teardown_test():
"""Power off all USB peripherals."""
# Disable main power first
psu = Net.get('VDD', type=NetType.PowerSupply)
psu.disable()
# Power off USB peripherals
for name in ['PROGRAMMER', 'SENSOR', 'DEBUGGER']:
try:
usb = Net.get(name, type=NetType.Usb)
usb.disable()
except Exception:
pass