Skip to content

Commit

Permalink
test(eda): add dataset test for report
Browse files Browse the repository at this point in the history
  • Loading branch information
jinglinpeng committed Apr 28, 2021
1 parent 2c60479 commit 0de5208
Showing 1 changed file with 12 additions and 0 deletions.
12 changes: 12 additions & 0 deletions dataprep/tests/eda/test_create_report.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
import pandas as pd
import pytest
from ...eda import create_report
from ...datasets import load_dataset

LOGGER = logging.getLogger(__name__)

Expand Down Expand Up @@ -83,3 +84,14 @@ def test_report_single_column(simpledf: pd.DataFrame) -> None:

matplotlib.use("PS")
create_report(simpledf[["a"]], mode="basic")


def test_dataset() -> None:
dataset_names = ["titanic", "iris"]
# dataset_names = get_dataset_names()
for dataset in dataset_names:
print(f"testing dataset:{dataset}")
df = load_dataset(dataset)
# popu_size = df.shape[0]
# df = df.sample(n=min(popu_size, 1000), random_state=0)
create_report(df)

0 comments on commit 0de5208

Please sign in to comment.