8th day of python challenges 111-117

venv/lib/python3.6/site-packages/pandas/tests/test_downstream.py

@@ -0,0 +1,172 @@
+"""
+Testing that we work in the downstream packages
+"""
+import importlib
+import subprocess
+import sys
+
+import numpy as np  # noqa
+import pytest
+
+from pandas.compat import PY36
+
+from pandas import DataFrame, Series
+from pandas.util import 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
+
+    if PY36:
+        try:
+            return importlib.import_module(name)
+        except ModuleNotFoundError:  # noqa
+            pytest.skip("skipping as {} not available".format(name))
+
+    else:
+        try:
+            return importlib.import_module(name)
+        except ImportError as e:
+            if "No module named" in str(e) and name in str(e):
+                pytest.skip("skipping as {} not available".format(name))
+            raise
+
+
+@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
+
+
+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")
+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:The 'warn':DeprecationWarning")
+@pytest.mark.filterwarnings("ignore:pandas.util:DeprecationWarning")
+@pytest.mark.filterwarnings("ignore:can't resolve:ImportWarning")
+@pytest.mark.skip(reason="gh-25778: geopandas stack issue")
+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")
+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