import mock
|
from data_manager.actions.predictions_to_annotations import predictions_to_annotations_form
|
from projects.models import Project
|
from users.models import User
|
|
|
def test_predictions_to_annotations_form():
|
project = Project()
|
user = User()
|
|
with mock.patch('projects.models.Project.get_model_versions') as mock_get_model_versions:
|
project.model_version = ''
|
mock_get_model_versions.return_value = ['undefined']
|
assert predictions_to_annotations_form(user, project)[0]['fields'][0]['options'] == ['undefined']
|
|
project.model_version = None
|
mock_get_model_versions.return_value = ['undefined']
|
assert predictions_to_annotations_form(user, project)[0]['fields'][0]['options'] == ['undefined']
|
|
project.model_version = 'undefined'
|
mock_get_model_versions.return_value = ['undefined']
|
assert predictions_to_annotations_form(user, project)[0]['fields'][0]['options'] == ['undefined']
|
|
project.model_version = ''
|
mock_get_model_versions.return_value = []
|
assert predictions_to_annotations_form(user, project)[0]['fields'][0]['options'] == []
|
|
project.model_version = None
|
mock_get_model_versions.return_value = []
|
assert predictions_to_annotations_form(user, project)[0]['fields'][0]['options'] == []
|
|
project.model_version = 'undefined'
|
mock_get_model_versions.return_value = []
|
assert predictions_to_annotations_form(user, project)[0]['fields'][0]['options'] == ['undefined']
|
