def test_custom_form_model_info(self, custom_form_model_properties):
model = _models.CustomFormModelInfo(
model_id=1, status=_models.CustomFormModelStatus.READY, training_started_on=datetime.datetime(1, 1, 1), training_completed_on=datetime.datetime(1, 1, 1),
properties=custom_form_model_properties[0], model_name="my model"
)
model_repr = "CustomFormModelInfo(model_id=1, status=ready, training_started_on=0001-01-01 00:00:00, training_completed_on=0001-01-01 00:00:00, properties={}, model_name=my model)".format(custom_form_model_properties[1])[:1024]
assert repr(model) == model_repr
def test_custom_form_model_info(self):
model = _models.CustomFormModelInfo(
model_id=1,
status=_models.CustomFormModelStatus.ready,
training_started_on=datetime.datetime(1, 1, 1),
training_completed_on=datetime.datetime(1, 1, 1))
model_repr = "CustomFormModelInfo(model_id=1, status=ready, training_started_on=0001-01-01 00:00:00, training_completed_on=0001-01-01 00:00:00)"[:
1024]
assert repr(model) == model_repr
def test_custom_form_model_info(self):
model = _models.CustomFormModelInfo(
model_id=1,
status=_models.CustomFormModelStatus.ready,
created_on=datetime.datetime(1, 1, 1),
last_modified=datetime.datetime(1, 1, 1))
model_repr = "CustomFormModelInfo(model_id=1, status=ready, created_on=0001-01-01 00:00:00, last_modified=0001-01-01 00:00:00)"[:
1024]
assert repr(model) == model_repr
def test_custom_form_model_info_to_dict(self):
model = _models.CustomFormModelInfo(
model_id="1234",
status="creating",
training_started_on=datetime(2021, 1, 10, 23, 55, 59, 342380),
training_completed_on=datetime(2021, 1, 10, 23, 55, 59, 342380),
model_name="sample_model",
properties=_models.CustomFormModelProperties(
is_composed_model=False, ))
d = model.to_dict()
final = {
"model_id": "1234",
"status": "creating",
"training_started_on": datetime(2021, 1, 10, 23, 55, 59, 342380),
"training_completed_on": datetime(2021, 1, 10, 23, 55, 59, 342380),
"model_name": "sample_model",
"properties": {
"is_composed_model": False,
}
}
assert d == final