add and mul operations do not issue a numexpr warning as of version 2.13.1.

Define NUMEXPR_VERSION in computation.check and use this to switch related tests on/off

Author: eoincondron

pandas/core/computation/check.py

@@ -1,8 +1,15 @@
 from __future__ import annotations
 
+import re
+
 from pandas.compat._optional import import_optional_dependency
 
 ne = import_optional_dependency("numexpr", errors="warn")
 NUMEXPR_INSTALLED = ne is not None
-
-__all__ = ["NUMEXPR_INSTALLED"]
+if NUMEXPR_INSTALLED:
+    try:
+        NUMEXPR_VERSION = tuple(map(int, re.findall(r"(\d+)", ne.__version__)))
+    except Exception:
+        NUMEXPR_VERSION = None
+else:
+    NUMEXPR_VERSION = None

pandas/tests/frame/test_arithmetic.py

@@ -20,6 +20,7 @@
 )
 import pandas._testing as tm
 from pandas.core.computation import expressions as expr
+from pandas.core.computation.check import NUMEXPR_VERSION
 from pandas.tests.frame.common import (
     _check_mixed_float,
     _check_mixed_int,
@@ -1095,9 +1096,12 @@ def test_binop_other(self, op, value, dtype, switch_numexpr_min_elements):
         skip = {
             (operator.truediv, "bool"),
             (operator.pow, "bool"),
-            (operator.add, "bool"),
-            (operator.mul, "bool"),
         }
+        if NUMEXPR_VERSION and NUMEXPR_VERSION <= (2, 13, 0):
+            skip |= {
+                (operator.add, "bool"),
+                (operator.mul, "bool"),
+            }
 
         elem = DummyElement(value, dtype)
         df = DataFrame({"A": [elem.value, elem.value]}, dtype=elem.dtype)

pandas/tests/series/test_arithmetic.py

@@ -25,7 +25,10 @@
 import pandas._testing as tm
 from pandas.core import ops
 from pandas.core.computation import expressions as expr
-from pandas.core.computation.check import NUMEXPR_INSTALLED
+from pandas.core.computation.check import (
+    NUMEXPR_INSTALLED,
+    NUMEXPR_VERSION,
+)
 
 
 @pytest.fixture(autouse=True, params=[0, 1000000], ids=["numexpr", "python"])
@@ -350,7 +353,9 @@ def test_add_list_to_masked_array_boolean(self, request):
         # GH#22962
         warning = (
             UserWarning
-            if request.node.callspec.id == "numexpr" and NUMEXPR_INSTALLED
+            if request.node.callspec.id == "numexpr"
+            and NUMEXPR_INSTALLED
+            and NUMEXPR_VERSION <= (2, 13, 0)
             else None
         )
         ser = Series([True, None, False], dtype="boolean")

pandas/tests/test_expressions.py

@@ -8,6 +8,7 @@
 import pandas._testing as tm
 from pandas.core.api import DataFrame
 from pandas.core.computation import expressions as expr
+from pandas.core.computation.check import NUMEXPR_VERSION
 
 
 @pytest.fixture
@@ -318,7 +319,7 @@ def test_bool_ops_raise_on_arithmetic(self, op_str, opname):
             f(df, True)
 
     @pytest.mark.parametrize(
-        "op_str,opname", [("+", "add"), ("*", "mul"), ("-", "sub")]
+        "op_str, opname", [("+", "add"), ("*", "mul"), ("-", "sub")]
     )
     def test_bool_ops_warn_on_arithmetic(self, op_str, opname, monkeypatch):
         n = 10
@@ -328,6 +329,8 @@ def test_bool_ops_warn_on_arithmetic(self, op_str, opname, monkeypatch):
                 "b": np.random.default_rng(2).random(n) > 0.5,
             }
         )
+        if NUMEXPR_VERSION and NUMEXPR_VERSION >= (2, 13, 1) and op_str in ("+", "*"):
+            return
 
         subs = {"+": "|", "*": "&", "-": "^"}
         sub_funcs = {"|": "or_", "&": "and_", "^": "xor"}