Fix logic for _run_test

apache · Sep 20, 2023 · 5d6bb03 · 5d6bb03
1 parent 603b517
commit 5d6bb03
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/sdks/python/apache_beam/dataframe/frames_test.py b/sdks/python/apache_beam/dataframe/frames_test.py
@@ -187,6 +187,9 @@ def _run_test(
         if expected.index.is_unique:
           expected = expected.sort_index()
           actual = actual.sort_index()
+        elif isinstance(expected, pd.Series):
+          expected = expected.sort_values()
+          actual = actual.sort_values()
         else:
           expected = expected.sort_values(list(expected.columns))
           actual = actual.sort_values(list(actual.columns))