Exemplo n.º 1
0
 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"
Exemplo n.º 2
0
 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
Exemplo n.º 3
0
 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)