From 1be7187abe8ab95b67a01baf596495351a1c4d4f Mon Sep 17 00:00:00 2001 From: Jerome Dockes Date: Fri, 17 Nov 2023 11:47:28 +0100 Subject: [PATCH] more tests --- skrub/_dataframe/tests/test_common.py | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 skrub/_dataframe/tests/test_common.py diff --git a/skrub/_dataframe/tests/test_common.py b/skrub/_dataframe/tests/test_common.py new file mode 100644 index 000000000..182a664e8 --- /dev/null +++ b/skrub/_dataframe/tests/test_common.py @@ -0,0 +1,11 @@ +from skrub._dataframe import skrubns, std, stdns + + +def test_std(px): + df = px.DataFrame({"A": [1, 2]}) + assert hasattr(std(df), "dataframe") + assert hasattr(stdns(df), "dataframe_from_columns") + ns = skrubns(df) + s = ns.make_series([1, 2], name="A") + assert hasattr(std(s), "column") + assert hasattr(stdns(s), "dataframe_from_columns")