f722d46a4950a1eb829608e0da6e91b49776401d,tests/test_estimators.py,TestEstimators,test_template_classifier,#TestEstimators#,20

Before Change

        check_estimator(TemplateTransformer)

    def test_template_classifier(self):
        check_estimator(TemplateClassifier)

After Change

        assert_allclose(X_trans, np.sqrt(X))

    def test_template_classifier(self):
        X, y = self.data
        clf = TemplateClassifier()
        assert clf.demo_param == "demo"

        clf.fit(X, y)
        assert hasattr(clf, "classes_")
        assert hasattr(clf, "X_")
        assert hasattr(clf, "y_")

        y_pred = clf.predict(X)
        assert y_pred.shape == (X.shape[0],)

Instances