from mlxtend.data import iris_data
from mlxtend.data import wine_data
from mlxtend.data import autompg_data
from mlxtend.data import mnist_data

def test_import_wine_data():
    X, y = wine_data()
    assert(X.shape[0] == 178)
    assert(X.shape[1] == 13)
    assert(y.shape[0] == 178)

def test_import_iris_data():
    X, y = iris_data()
    assert(X.shape[0] == 150)
    assert(X.shape[1] == 4)
    assert(y.shape[0] == 150)

def test_import_autompg_data():
    X, y = autompg_data()
    assert(X.shape[0] == 392)
    assert(X.shape[1] == 8)
    assert(y.shape[0] == 392)
    
def test_import_mnist_data():
    X, y = mnist_data()
    assert(X.shape[0] == 5000)
    assert(X.shape[1] == 784)
    assert(y.shape[0] == 5000)
from mlxtend.data import iris_data
from mlxtend.data import wine_data
from mlxtend.data import autompg_data
from mlxtend.data import mnist_data

def test_import_wine_data():
    X, y = wine_data()
    assert(X.shape[0] == 178)
    assert(X.shape[1] == 13)
    assert(y.shape[0] == 178)

def test_import_iris_data():
    X, y = iris_data()
    assert(X.shape[0] == 150)
    assert(X.shape[1] == 4)
    assert(y.shape[0] == 150)

def test_import_autompg_data():
    X, y = autompg_data()
    assert(X.shape[0] == 392)
    assert(X.shape[1] == 8)
    assert(y.shape[0] == 392)
    
#def test_import_mnist_data():
#    X, y = mnist_data()
#    assert(X.shape[0] == 5000)
#    assert(X.shape[1] == 784)
#    assert(y.shape[0] == 5000)
exclude mnist from travis