Steps

from tofupilot import TofuPilotClient
from datetime import datetime

client = TofuPilotClient()

# Simple true/false step
def step_connect():
    return True,

# Step with a measurement value and unit
def step_initial_charge_check():
    initial_charge = 80
    return True, initial_charge, "%"

# Step with a measurement and a lower limit
def step_initial_temp_check():
    initial_temp = 72.0
    limit_low = 0.0
    return initial_temp >= limit_low, initial_temp, "°C", limit_low

# Step with a measurement, lower limit, and upper limit
def step_temp_calibration():
    calibrated_temp = 75.0
    limit_low = 70.0
    limit_high = 80.0
    return limit_low <= calibrated_temp <= limit_high, calibrated_temp, "°C", limit_low, limit_high

# Simple step sequencer and logger
def run_all_tests():
    tests = [func for name, func in globals().items() if callable(func) and name.startswith("step_")]
    steps, all_passed = [], True

    for t in tests:
        start = datetime.now()
        passed, value, unit, low, high = (t() + (None,) * 4)[:5]  # unpack with default None
        end = datetime.now()
        steps.append({
            "name": t.__name__,
            "started_at": start,
            "duration": end - start,
            "step_passed": passed,
            "measurement_value": value,
            "measurement_unit": unit,
            "limit_low": low,
            "limit_high": high
        })
        all_passed &= passed

    return all_passed, steps

run_passed, steps = run_all_tests()

client.create_run(
    procedure_id="FVT1",
    unit_under_test={
      "serial_number": "00102",
    },
    run_passed=run_passed,
    steps=steps,
)