def test_fit_returns_self(self): """Tests if the `fit()` method returns `self`. This allows to quickly implement one-liners with AutoML""" model = AutoML() model.results_path = self.automl_dir self.assertTrue( isinstance(model.fit(iris.data, iris.target), AutoML), "`fit()` method must return 'self'", )
def test_no_constructor_args(self): """Tests the use of AutoML without passing any args. Should work without any arguments""" # Create model with no arguments model = AutoML() model.results_path = self.automl_dir # Assert than an Exception is raised score = model.fit(iris.data, iris.target).score(iris.data, iris.target) self.assertGreater(score, 0.5)