def test_create_duplicate_classifier_model(self): try: create_classifier_model(1, self.data, self.classifier_class) except Exception as e: assert "already exists" in e.args[0].lower() else: assert False, "Model with same version should not be created"
def test_valid_classifier_model(self): classifier_model = create_classifier_model(2, self.data, self.classifier_class) assert type(classifier_model) == ClassifierModel assert classifier_model.id is None classifier = classifier_model.classifier assert type(classifier) == self.classifier_class
def _get_model(self, version): # first create classifier csv_path = 'fixtures/processed_data_for_testing.csv' data = get_processed_data(csv_path) return create_classifier_model(version, data)