166 lines
4.6 KiB
Python
166 lines
4.6 KiB
Python
"""
|
|
Testing that we work in the downstream packages
|
|
"""
|
|
import importlib
|
|
import subprocess
|
|
import sys
|
|
|
|
import numpy as np # noqa
|
|
import pytest
|
|
|
|
from pandas import DataFrame, Series
|
|
import pandas._testing as tm
|
|
|
|
|
|
def import_module(name):
|
|
# we *only* want to skip if the module is truly not available
|
|
# and NOT just an actual import error because of pandas changes
|
|
|
|
try:
|
|
return importlib.import_module(name)
|
|
except ModuleNotFoundError: # noqa
|
|
pytest.skip("skipping as {} not available".format(name))
|
|
|
|
|
|
@pytest.fixture
|
|
def df():
|
|
return DataFrame({"A": [1, 2, 3]})
|
|
|
|
|
|
def test_dask(df):
|
|
|
|
toolz = import_module("toolz") # noqa
|
|
dask = import_module("dask") # noqa
|
|
|
|
import dask.dataframe as dd
|
|
|
|
ddf = dd.from_pandas(df, npartitions=3)
|
|
assert ddf.A is not None
|
|
assert ddf.compute() is not None
|
|
|
|
|
|
@pytest.mark.filterwarnings("ignore:Panel class is removed")
|
|
def test_xarray(df):
|
|
|
|
xarray = import_module("xarray") # noqa
|
|
|
|
assert df.to_xarray() is not None
|
|
|
|
|
|
def test_oo_optimizable():
|
|
# GH 21071
|
|
subprocess.check_call([sys.executable, "-OO", "-c", "import pandas"])
|
|
|
|
|
|
@tm.network
|
|
# Cython import warning
|
|
@pytest.mark.filterwarnings("ignore:can't:ImportWarning")
|
|
@pytest.mark.filterwarnings(
|
|
# patsy needs to update their imports
|
|
"ignore:Using or importing the ABCs from 'collections:DeprecationWarning"
|
|
)
|
|
def test_statsmodels():
|
|
|
|
statsmodels = import_module("statsmodels") # noqa
|
|
import statsmodels.api as sm
|
|
import statsmodels.formula.api as smf
|
|
|
|
df = sm.datasets.get_rdataset("Guerry", "HistData").data
|
|
smf.ols("Lottery ~ Literacy + np.log(Pop1831)", data=df).fit()
|
|
|
|
|
|
# Cython import warning
|
|
@pytest.mark.filterwarnings("ignore:can't:ImportWarning")
|
|
def test_scikit_learn(df):
|
|
|
|
sklearn = import_module("sklearn") # noqa
|
|
from sklearn import svm, datasets
|
|
|
|
digits = datasets.load_digits()
|
|
clf = svm.SVC(gamma=0.001, C=100.0)
|
|
clf.fit(digits.data[:-1], digits.target[:-1])
|
|
clf.predict(digits.data[-1:])
|
|
|
|
|
|
# Cython import warning and traitlets
|
|
@tm.network
|
|
@pytest.mark.filterwarnings("ignore")
|
|
def test_seaborn():
|
|
|
|
seaborn = import_module("seaborn")
|
|
tips = seaborn.load_dataset("tips")
|
|
seaborn.stripplot(x="day", y="total_bill", data=tips)
|
|
|
|
|
|
def test_pandas_gbq(df):
|
|
|
|
pandas_gbq = import_module("pandas_gbq") # noqa
|
|
|
|
|
|
@pytest.mark.xfail(reason="0.7.0 pending")
|
|
@tm.network
|
|
def test_pandas_datareader():
|
|
|
|
pandas_datareader = import_module("pandas_datareader") # noqa
|
|
pandas_datareader.DataReader("F", "quandl", "2017-01-01", "2017-02-01")
|
|
|
|
|
|
# importing from pandas, Cython import warning
|
|
@pytest.mark.filterwarnings("ignore:can't resolve:ImportWarning")
|
|
@pytest.mark.skip(reason="Anaconda installation issue - GH32144")
|
|
def test_geopandas():
|
|
|
|
geopandas = import_module("geopandas") # noqa
|
|
fp = geopandas.datasets.get_path("naturalearth_lowres")
|
|
assert geopandas.read_file(fp) is not None
|
|
|
|
|
|
def test_geopandas_coordinate_indexer():
|
|
# this test is included to have coverage of one case in the indexing.py
|
|
# code that is only kept for compatibility with geopandas, see
|
|
# https://github.com/pandas-dev/pandas/issues/27258
|
|
# We should be able to remove this after some time when its usage is
|
|
# removed in geopandas
|
|
from pandas.core.indexing import _NDFrameIndexer
|
|
|
|
class _CoordinateIndexer(_NDFrameIndexer):
|
|
def _getitem_tuple(self, tup):
|
|
obj = self.obj
|
|
xs, ys = tup
|
|
return obj[xs][ys]
|
|
|
|
Series._create_indexer("cx", _CoordinateIndexer)
|
|
s = Series(range(5))
|
|
res = s.cx[:, :]
|
|
tm.assert_series_equal(s, res)
|
|
|
|
|
|
# Cython import warning
|
|
@pytest.mark.filterwarnings("ignore:can't resolve:ImportWarning")
|
|
@pytest.mark.filterwarnings("ignore:RangeIndex.* is deprecated:DeprecationWarning")
|
|
def test_pyarrow(df):
|
|
|
|
pyarrow = import_module("pyarrow") # noqa
|
|
table = pyarrow.Table.from_pandas(df)
|
|
result = table.to_pandas()
|
|
tm.assert_frame_equal(result, df)
|
|
|
|
|
|
@pytest.mark.xfail(reason="pandas-wheels-50", strict=False)
|
|
def test_missing_required_dependency():
|
|
# GH 23868
|
|
# To ensure proper isolation, we pass these flags
|
|
# -S : disable site-packages
|
|
# -s : disable user site-packages
|
|
# -E : disable PYTHON* env vars, especially PYTHONPATH
|
|
# And, that's apparently not enough, so we give up.
|
|
# https://github.com/MacPython/pandas-wheels/pull/50
|
|
call = ["python", "-sSE", "-c", "import pandas"]
|
|
|
|
with pytest.raises(subprocess.CalledProcessError) as exc:
|
|
subprocess.check_output(call, stderr=subprocess.STDOUT)
|
|
|
|
output = exc.value.stdout.decode()
|
|
for name in ["numpy", "pytz", "dateutil"]:
|
|
assert name in output
|