diff --git a/python-sdk/tests_integration/sql/operators/test_load_file.py b/python-sdk/tests_integration/sql/operators/test_load_file.py index dacebc9d46..189ca6e5bb 100644 --- a/python-sdk/tests_integration/sql/operators/test_load_file.py +++ b/python-sdk/tests_integration/sql/operators/test_load_file.py @@ -337,7 +337,9 @@ def test_aql_load_file_local_file_pattern_dataframe(sample_dag): def validate(input_df): assert isinstance(input_df, pd.DataFrame) assert test_df.shape == input_df.shape - assert test_df.sort_values("sell").equals(input_df.sort_values("sell")) + assert test_df.sort_values("sell", ignore_index=True).equals( + input_df.sort_values("sell", ignore_index=True) + ) print(input_df) with sample_dag: