Update unit test

thicket/tests/test_columnar_join.py

@@ -5,8 +5,9 @@
 
 import re
 
-import hatchet as ht
+import pandas as pd
 
+import hatchet as ht
 from test_filter_metadata import filter_one_column
 from test_filter_metadata import filter_multiple_and
 from test_filter_stats import check_filter_stats
@@ -47,6 +48,11 @@ def test_columnar_join(columnar_join_thicket):
         [len(th.profile_mapping) for th in thicket_list]
     )
 
+    # PerfData and StatsFrame nodes should be in the same order.
+    assert (pd.unique(combined_th.dataframe.reset_index()["node"].tolist()) == pd.unique(
+        combined_th.statsframe.dataframe.reset_index()["node"].tolist())
+    ).all()
+
 
 def test_filter_columnar_join(columnar_join_thicket):
     thicket_list, thicket_list_cp, combined_th = columnar_join_thicket