sorting by location then count

ROCm · Jul 13, 2023 · 94554b4 · 94554b4
1 parent deb2569
commit 94554b4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/source/python/gui/source/tests/test_graphs.py b/source/python/gui/source/tests/test_graphs.py
@@ -217,8 +217,9 @@ def test_parse_files_default():
     ][:2].round(4)
 
     assert sorted(file_names_run) == sorted(file_names)
+    _samples_df_expected_counts = [138, 152, 276, 138, 152, 828, 138, 152, 138]
 
-    samples_df = samples_df.sort_values(by="location")
+    samples_df = samples_df.sort_values(by=["location","count"])
     samples_df_locations = pd.concat(
         [samples_df[0:3], samples_df[100:103], samples_df[150:153]]
     )["location"].to_numpy()
@@ -227,7 +228,7 @@ def test_parse_files_default():
     )["count"].to_numpy()
 
     assert (samples_df_locations == samples_df_expected_locations).all()
-    assert (samples_df_counts == samples_df_expected_counts).all()
+    assert (samples_df_counts == _samples_df_expected_counts).all()
 
     # assert expected speedup err
     assert (top_df["program speedup"].to_numpy() == top_df_expected_program_speedup).all()