Add pandas dataframe and series bool checker

Author: Peter Hamfelt

pylint_ml/checkers/pandas/pandas_dataframe_bool.py

@@ -0,0 +1,38 @@
+# Licensed under the MIT: https://mit-license.org/
+# For details: https://github.com/pylint-dev/pylint-ml/LICENSE
+# Copyright (c) https://github.com/pylint-dev/pylint-ml/CONTRIBUTORS.txt
+
+"""Check for usage of the deprecated pandas DataFrame.bool() method."""
+
+from __future__ import annotations
+
+from astroid import nodes
+from pylint.checkers import BaseChecker
+from pylint.checkers.utils import only_required_for_messages
+from pylint.interfaces import HIGH
+
+
+class PandasDataFrameBoolChecker(BaseChecker):
+    name = "pandas-dataframe-bool"
+    msgs = {
+        "W8104": (
+            "Use of deprecated pandas DataFrame.bool() method",
+            "pandas-dataframe-bool",
+            "Avoid using the deprecated pandas DataFrame.bool() method.",
+        ),
+    }
+
+    @only_required_for_messages("pandas-dataframe-bool")
+    def visit_call(self, node: nodes.Call) -> None:
+        if isinstance(node.func, nodes.Attribute):
+            method_name = getattr(node.func, "attrname", None)
+            module_name = getattr(node.func.expr, "name", None)
+
+            if method_name == "bool" and module_name == "pd":
+                self.add_message("pandas-dataframe-bool", node=node, confidence=HIGH)
+
+    def _check_method_usage(self, node):
+        method_name = getattr(node.func, "attrname", None)
+        module_name = getattr(node.func.expr, "name", None)
+        if method_name == "bool" and module_name == "pd":
+            self.add_message("pandas-dataframe-bool", node=node, confidence=HIGH)

pylint_ml/checkers/pandas/pandas_series_bool.py

@@ -0,0 +1,38 @@
+# Licensed under the MIT: https://mit-license.org/
+# For details: https://github.com/pylint-dev/pylint-ml/LICENSE
+# Copyright (c) https://github.com/pylint-dev/pylint-ml/CONTRIBUTORS.txt
+
+"""Check for usage of the deprecated pandas Series.bool() method."""
+
+from __future__ import annotations
+
+from astroid import nodes
+from pylint.checkers import BaseChecker
+from pylint.checkers.utils import only_required_for_messages
+from pylint.interfaces import HIGH
+
+
+class PandasSeriesBoolChecker(BaseChecker):
+    name = "pandas-series-bool"
+    msgs = {
+        "W8105": (
+            "Use of deprecated pandas Series.bool() method",
+            "pandas-series-bool",
+            "Avoid using the deprecated pandas Series.bool() method.",
+        ),
+    }
+
+    @only_required_for_messages("pandas-series-bool")
+    def visit_call(self, node: nodes.Call) -> None:
+        if isinstance(node.func, nodes.Attribute):
+            method_name = getattr(node.func, "attrname", None)
+            module_name = getattr(node.func.expr, "name", None)
+
+            if method_name == "bool" and module_name == "pd":
+                self.add_message("pandas-series-bool", node=node, confidence=HIGH)
+
+    def _check_method_usage(self, node):
+        method_name = getattr(node.func, "attrname", None)
+        module_name = getattr(node.func.expr, "name", None)
+        if method_name == "bool" and module_name == "pd":
+            self.add_message("pandas-series-bool", node=node, confidence=HIGH)

tests/checkers/test_pandas/test_pandas_dataframe_bool.py

@@ -0,0 +1,37 @@
+import astroid
+import pylint.testutils
+from pylint.interfaces import HIGH
+
+from pylint_ml.checkers.pandas.pandas_dataframe_bool import PandasDataFrameBoolChecker
+
+
+class TestDataFrameBoolChecker(pylint.testutils.CheckerTestCase):
+    CHECKER_CLASS = PandasDataFrameBoolChecker
+
+    def test_dataframe_bool_usage(self):
+        node = astroid.extract_node(
+            """
+            import pandas as pd
+            df_customers = pd.DataFrame(data)
+            df_customers.bool()  # [pandas-dataframe-bool]
+            """
+        )
+        with self.assertAddsMessages(
+            pylint.testutils.MessageTest(
+                msg_id="pandas-dataframe-bool",
+                confidence=HIGH,
+                node=node,
+            )
+        ):
+            self.checker.visit_call(node)
+
+    def test_no_bool_usage(self):
+        node = astroid.extract_node(
+            """
+            import pandas as pd
+            df_customers = pd.DataFrame(data)
+            df_customers.sum()  # This should pass without warnings
+            """
+        )
+        with self.assertNoMessages():
+            self.checker.visit_call(node)

tests/checkers/test_pandas/test_pandas_series_bool.py

@@ -0,0 +1,37 @@
+import astroid
+import pylint.testutils
+from pylint.interfaces import HIGH
+
+from pylint_ml.checkers.pandas.pandas_series_bool import PandasSeriesBoolChecker
+
+
+class TestSeriesBoolChecker(pylint.testutils.CheckerTestCase):
+    CHECKER_CLASS = PandasSeriesBoolChecker
+
+    def test_series_bool_usage(self):
+        node = astroid.extract_node(
+            """
+            import pandas as pd
+            series = pd.Series(data)
+            series.bool()  # [pandas-series-bool]
+            """
+        )
+        with self.assertAddsMessages(
+            pylint.testutils.MessageTest(
+                msg_id="pandas-series-bool",
+                confidence=HIGH,
+                node=node,
+            )
+        ):
+            self.checker.visit_call(node)
+
+    def test_no_bool_usage(self):
+        node = astroid.extract_node(
+            """
+            import pandas as pd
+            series = pd.Series(data)
+            series.sum()  # This should pass without warnings
+            """
+        )
+        with self.assertNoMessages():
+            self.checker.visit_call(node)