When comparing Series, sort the values in Dataframe tests (#28557)

apache · Sep 20, 2023 · f676d93 · f676d93
1 parent 93de970
commit f676d93
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
@@ -193,6 +193,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))