def test_merge_metrics_model():
metrics = ModelMetrics()
other = ModelMetrics(model_type=ModelType.REGRESSION)
other.regression_metrics = None
new_metrics = metrics.merge(other)
assert new_metrics.model_type == ModelType.REGRESSION
assert new_metrics.confusion_matrix is None
# keep initial model type during merge
metrics = ModelMetrics(model_type=ModelType.REGRESSION)
other = ModelMetrics(model_type=ModelType.CLASSIFICATION)
other.regression_metrics = None
new_metrics = metrics.merge(other)
assert new_metrics.model_type == ModelType.REGRESSION
assert new_metrics.confusion_matrix is None
def test_merge_metrics_with_none_confusion_matrix():
metrics = ModelMetrics()
other = ModelMetrics()
other.confusion_matrix = None
other.regression_metrics = None
new_metrics = metrics.merge(other)
assert new_metrics.model_type == ModelType.UNKNOWN
def test_merge_metrics_with_none_regression_matrix():
metrics = ModelMetrics()
other = ModelMetrics(model_type=ModelType.REGRESSION)
other.regression_metrics = None
new_metrics = metrics.merge(other)
assert new_metrics.model_type == ModelType.REGRESSION
