diff --git a/tests/component/task/test_triggers.py b/tests/component/task/test_triggers.py index 67dcdf56..7357aabf 100644 --- a/tests/component/task/test_triggers.py +++ b/tests/component/task/test_triggers.py @@ -92,6 +92,7 @@ def test___start_trigger___wait_for_valid_timestamp___no_errors( ai_voltage_task.start() timestamp = ai_voltage_task.wait_for_valid_timestamp(TimestampEvent.START_TRIGGER) + assert isinstance(timestamp, ht_datetime) @@ -103,6 +104,7 @@ def test___reference_trigger___wait_for_valid_timestamp___no_errors( ai_voltage_task.start() timestamp = ai_voltage_task.wait_for_valid_timestamp(TimestampEvent.REFERENCE_TRIGGER) + assert isinstance(timestamp, ht_datetime) @@ -115,6 +117,7 @@ def test___arm_start_trigger___wait_for_valid_timestamp___no_errors( ci_count_edges_task.start() timestamp = ci_count_edges_task.wait_for_valid_timestamp(TimestampEvent.ARM_START_TRIGGER) + assert isinstance(timestamp, ht_datetime)