Skip to content

Commit 76175ee

Browse files
committed
fix profiler test
1 parent 0ff9762 commit 76175ee

1 file changed

Lines changed: 11 additions & 15 deletions

File tree

tests/fast/test_profiler.py

Lines changed: 11 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -20,23 +20,19 @@ def test_profiler_matches_expected_format(self, profiling_connection, tmp_path_f
2020
profiling_info_json = profiling_info.to_json()
2121
assert isinstance(profiling_info_json, str)
2222

23-
# Test expected metrics are there and profiling is json loadable
23+
# Test expected metrics are there and profiling is json loadable. The profiling output is now grouped
24+
# into top-level sections (the flat per-metric keys moved underneath these, e.g. latency -> query.total_time,
25+
# total_bytes_read -> io.total_bytes_read, system_peak_buffer_memory -> system.peak_buffer_memory).
2426
profiling_dict = profiling_info.to_pydict()
2527
expected_keys = {
26-
"query_name",
27-
"total_bytes_written",
28-
"total_bytes_read",
29-
"system_peak_temp_dir_size",
30-
"system_peak_buffer_memory",
31-
"rows_returned",
32-
"result_set_size",
33-
"latency",
34-
"cumulative_rows_scanned",
35-
"cumulative_cardinality",
36-
"cpu_time",
37-
"extra_info",
38-
"blocked_thread_time",
39-
"children",
28+
"query",
29+
"system",
30+
"io",
31+
"operator",
32+
"optimizer",
33+
"physical_planner",
34+
"planner",
35+
"parser",
4036
}
4137
assert expected_keys.issubset(profiling_dict.keys())
4238

0 commit comments

Comments
 (0)