File size: 872 Bytes
7885a28 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import pytest
from pandas.compat._optional import VERSIONS
import pandas as pd
from pandas.core.computation import expr
from pandas.core.computation.engines import ENGINES
from pandas.util.version import Version
def test_compat():
# test we have compat with our version of numexpr
from pandas.core.computation.check import NUMEXPR_INSTALLED
ne = pytest.importorskip("numexpr")
ver = ne.__version__
if Version(ver) < Version(VERSIONS["numexpr"]):
assert not NUMEXPR_INSTALLED
else:
assert NUMEXPR_INSTALLED
@pytest.mark.parametrize("engine", ENGINES)
@pytest.mark.parametrize("parser", expr.PARSERS)
def test_invalid_numexpr_version(engine, parser):
if engine == "numexpr":
pytest.importorskip("numexpr")
a, b = 1, 2 # noqa: F841
res = pd.eval("a + b", engine=engine, parser=parser)
assert res == 3
|