import numpy as np
import pandas as pd
import nose
import pytest
import statsmodels.datasets
from statsmodels.datasets.utils import Dataset
exclude = [
'check_internet', 'clear_data_home', 'get_data_home', 'get_rdataset',
'tests', 'utils', 'webuse'
]
datasets = []
for dataset_name in dir(statsmodels.datasets):
if not dataset_name.startswith('_') and dataset_name not in exclude:
datasets.append(dataset_name)
# TODO: Remove nottest when nose support is dropped
@nose.tools.nottest
@pytest.mark.parametrize('dataset_name', datasets)
def test_dataset(dataset_name):
dataset = importlib.import_module('statsmodels.datasets.' + dataset_name)
data = dataset.load()
assert isinstance(data, Dataset)
assert isinstance(data.data, np.recarray)
df_data = dataset.load_pandas()
assert isinstance(df_data, Dataset)
assert isinstance(df_data.data, pd.DataFrame)
import importlib
import numpy as np
import pandas as pd
import pytest
import statsmodels.datasets
from statsmodels.compat import PY3
from statsmodels.datasets.utils import Dataset
exclude = ['check_internet', 'clear_data_home', 'get_data_home',
'get_rdataset', 'tests', 'utils', 'webuse']
datasets = []
for dataset_name in dir(statsmodels.datasets):
if not dataset_name.startswith('_') and dataset_name not in exclude:
datasets.append(dataset_name)
@pytest.mark.parametrize('dataset_name', datasets)
def test_dataset(dataset_name):
dataset = importlib.import_module('statsmodels.datasets.' + dataset_name)
warning_type = FutureWarning if PY3 else None
with pytest.warns(warning_type):
ds = dataset.load()
assert isinstance(ds, Dataset)
assert isinstance(ds.data, np.recarray)
if hasattr(ds, 'exog'):
assert isinstance(ds.exog, np.ndarray)
if hasattr(ds, 'endog'):
assert isinstance(ds.endog, np.ndarray)