test is_numeric

zdelrosario · Jan 3, 2023 · 01c05a3 · 01c05a3
1 parent 9ab6956
commit 01c05a3
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_dfply_select.py b/tests/test_dfply_select.py
@@ -295,6 +295,16 @@ def test_select_if(self):
                 >> gr.tf_select_if(lambda col: any(col.str.contains(".")))
             )
         )
+        # test 6: is_numeric helper
+        self.assertEqual(
+            {"carat", "depth", "table", "price", "x", "y", "z"},
+            set(
+                (
+                    data.df_diamonds
+                    >> gr.tf_select_if(gr.is_numeric)
+                ).columns
+            )
+        )
 
     def test_drop_if(self):
         # test 1: returns a dataframe where any column does not have a mean greater than 3