class TestSteelsClassifier: def __init__(self): self.classifier = None def setup(self): self.classifier = SteelsClassifier() def test_classifier_1(self): """ Simple test using irises. """ iris = datasets.load_iris() self.classifier.fit(iris.data, iris.target) success = 0 for i, sample in enumerate(iris.data): success += self.classifier.predict([sample]) == iris.target[i] print(float(success)/len(iris.data)) def test_classifier_2(self): """ Simple test using easy learning data. """ self.classifier.fit([[1], [2], [3], [4]], [1, 2, 3, 4]) success = 0 for i in range(1, 5): success += self.classifier.predict([[i]]) == [i] print(float(success)/4)
def setup(self): self.classifier = SteelsClassifier()