def test_automl_build_old(old_auto_ml_predictor_data, basic_predictor_report_data):
session = mock.Mock()
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = pc.build(old_auto_ml_predictor_data)
assert predictor.name == 'AutoML predictor'
assert predictor.description == 'Predicts z from input x'
def test_build(valid_simple_ml_predictor_data, basic_predictor_report_data):
session = mock.Mock()
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = pc.build(valid_simple_ml_predictor_data)
assert predictor.name == 'ML predictor'
assert predictor.description == 'Predicts z from input x and latent variable y'
def test_graph_build(valid_graph_predictor_data, basic_predictor_report_data):
session = mock.Mock()
session.get_resource.return_value = basic_predictor_report_data
pc = PredictorCollection(uuid.uuid4(), session)
predictor = pc.build(valid_graph_predictor_data)
assert predictor.name == 'Graph predictor'
assert predictor.description == 'description'
assert len(predictor.predictors) == 2
assert len(predictor.training_data) == 1
def test_build(valid_simple_ml_data):
session = mock.Mock()
session.get_resource.return_value = {
'id': str(uuid.uuid4()),
'status': 'VALID',
'report': {}
}
pc = PredictorCollection(uuid.uuid4(), session)
predictor = pc.build(valid_simple_ml_data)
assert predictor.name == 'ML predictor'
assert predictor.description == 'Predicts z from input x and latent variable y'