import unittest

from sklearn import datasets

from sklearn.ensemble import RandomForestClassifier

from sklearn.linear_model import LinearRegression

class TestSklearn(unittest.TestCase):

def test_random_forest_classifier(self):

iris = datasets.load_iris()

X, y = iris.data, iris.target

rf1 = RandomForestClassifier()

rf1.fit(X,y)

def test_linearn_classifier(self):

iris = datasets.load_iris()

X, y = iris.data, iris.target

lr1 = LinearRegression()

lr1.fit(X,y)