From 4117f85680a3d85d5203da05f612f3cd4ea55819 Mon Sep 17 00:00:00 2001 From: Sweta Kumari Date: Sat, 3 Oct 2026 03:06:21 +0530 Subject: [PATCH] test(runs): add assertions to run function smoke checks --- tests/test_runs/test_run_functions.py | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) diff --git a/tests/test_runs/test_run_functions.py b/tests/test_runs/test_run_functions.py index cb5d5e59d..08fd04205 100644 --- a/tests/test_runs/test_run_functions.py +++ b/tests/test_runs/test_run_functions.py @@ -339,13 +339,16 @@ def _remove_random_state(flow): assert run_ == run assert isinstance(run.dataset_id, int) - # This is only a smoke check right now - # TODO add a few asserts here - run._to_xml() + # Check XML output. + xml = run._to_xml() + assert xml is not None + if run.trace is not None: - # This is only a smoke check right now - # TODO add a few asserts here - run.trace.trace_to_arff() + # Check trace ARFF output. + trace_arff = run.trace.trace_to_arff() + assert isinstance(trace_arff, dict) + assert "data" in trace_arff + assert len(trace_arff["data"]) > 0 # check arff output assert len(run.data_content) == num_instances