def test_failed_register(valid_simple_ml_data):
session = mock.Mock()
session.post_resource.side_effect = NotFound("/projects/uuid/not_found")
pc = PredictorCollection(uuid.uuid4(), session)
predictor = SimpleMLPredictor.build(valid_simple_ml_data)
with pytest.raises(ModuleRegistrationFailedException) as e:
pc.register(predictor)
assert 'The "SimpleMLPredictor" failed to register. NotFound: /projects/uuid/not_found' in str(e.value)
def test_register_experimental(valid_simple_ml_predictor_data, basic_predictor_report_data):
session = mock.Mock()
post_response = deepcopy(valid_simple_ml_predictor_data)
post_response["experimental"] = True
post_response["experimental_reasons"] = ["This is a test", "Of experimental reasons"]
session.post_resource.return_value = post_response
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = SimpleMLPredictor.build(valid_simple_ml_predictor_data)
with pytest.warns(UserWarning) as record:
pc.register(predictor)
msg = str(record[0].message)
assert "Predictor" in msg
assert "This is a test" in msg
assert "Of experimental reasons" in msg
def test_register(valid_simple_ml_predictor_data, basic_predictor_report_data):
session = mock.Mock()
session.post_resource.return_value = valid_simple_ml_predictor_data
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = SimpleMLPredictor.build(valid_simple_ml_predictor_data)
registered = pc.register(predictor)
assert registered.name == 'ML predictor'
def test_automl_register_old(old_auto_ml_predictor_data, basic_predictor_report_data):
session = mock.Mock()
session.post_resource.return_value = old_auto_ml_predictor_data
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = AutoMLPredictor.build(old_auto_ml_predictor_data)
registered = pc.register(predictor)
assert registered.name == 'AutoML predictor'
def test_graph_register(valid_graph_predictor_data, basic_predictor_report_data):
copy_graph_data = deepcopy(valid_graph_predictor_data)
session = mock.Mock()
session.post_resource.return_value = copy_graph_data
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = GraphPredictor.build(valid_graph_predictor_data)
registered = pc.register(predictor)
assert registered.name == 'Graph predictor'
def test_register(valid_simple_ml_data):
session = mock.Mock()
session.post_resource.return_value = valid_simple_ml_data
session.get_resource.return_value = {
'id': str(uuid.uuid4()),
'status': 'VALID',
'report': {}
}
pc = PredictorCollection(uuid.uuid4(), session)
predictor = SimpleMLPredictor.build(valid_simple_ml_data)
registered = pc.register(predictor)
assert registered.name == 'ML predictor'