Skip to content

Commit

Permalink
Use polars assert_frame_equal
Browse files Browse the repository at this point in the history
  • Loading branch information
jbenjoseph committed Nov 6, 2024
1 parent 78a6fba commit d082f53
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions tests/bindingdb_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from pathlib import Path
from unittest.mock import patch
import polars as pl
from polars.testing import assert_frame_equal
from aiondata import BindingDB
from aiondata import BindingAffinity

Expand Down Expand Up @@ -73,9 +74,7 @@ def test_dataframe_creation_from_cache(
mock_mkdir.assert_called_once_with(parents=True, exist_ok=True)
mock_write_parquet.assert_not_called()
mock_read_parquet.assert_called_once()
assert df.frame_equal(
mock_df
), "DataFrame content does not match expected mock DataFrame."
assert_frame_equal(df, mock_df)


@patch("pathlib.Path.mkdir")
Expand Down

0 comments on commit d082f53

Please sign in to comment.