Skip to content

Commit

Permalink
debugging
Browse files Browse the repository at this point in the history
Signed-off-by: Sajid Alam <sajid_alam@mckinsey.com>
  • Loading branch information
SajidAlamQB committed Jun 5, 2024
1 parent e552d1f commit 1155903
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions kedro-datasets/tests/databricks/test_managed_table_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -485,7 +485,14 @@ def test_load_pandas(self, sample_pandas_df: pd.DataFrame):
database="test", table="test_load_pandas", dataframe_type="pandas"
)
pandas_df = pandas_ds.load().sort_values("name", ignore_index=True)
expected_df = sample_pandas_df.sort_values("name", ignore_index=True)

print("Loaded DataFrame:")
print(pandas_df)
print("Expected DataFrame:")
print(expected_df)

assert isinstance(pandas_df, pd.DataFrame)
assert pandas_df.equals(
expected_df), f"DataFrames are not equal.\nLoaded DataFrame:\n{pandas_df}\nExpected DataFrame:\n{expected_df}"

assert isinstance(pandas_df, pd.DataFrame) and pandas_df.equals(
sample_pandas_df
)

0 comments on commit 1155903

Please sign in to comment.