Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Feb 6, 2024
1 parent 1b1c689 commit 24039f1
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/test_loaders.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,11 +211,12 @@ def test_synth_ae():
assert x_df.shape == (8, 16), f"x_df shape is {x_df.shape}"
assert y_df.shape == (8, 1)


def test_rdmp():
"""The RDMP dataloader."""
x_df, y_df = loaders.get_data_sklearn("RDMP", DATA_FOLDER)

assert 'death' not in x_df.columns
assert 'age' in x_df.columns
assert "death" not in x_df.columns
assert "age" in x_df.columns

assert y_df.shape[1] == 1

0 comments on commit 24039f1

Please sign in to comment.