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_confusion_matrix(): metrics = ModelMetrics() other = ModelMetrics() other.confusion_matrix = None metrics.merge(other)
def test_merge_none(): metrics = ModelMetrics() assert metrics.merge(None) == metrics
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
def test_merge_none(): metrics = ModelMetrics() metrics.merge(None)