Skip to content
Snippets Groups Projects
Commit b5f6ca89 authored by Erwan Rouchet's avatar Erwan Rouchet
Browse files

Merge branch 'no-mltool' into 'master'

Remove MLTool unique import

See merge request !1031
parents cb83de03 0d13d154
No related branches found
No related tags found
1 merge request!1031Remove MLTool unique import
from unittest.mock import MagicMock
from django.test import TestCase
from arkindex_common.ml_tool import MLTool, MLToolType
from arkindex_common.ml_tool import MLToolType
from arkindex.documents.models import DataSource
class TestDataSource(TestCase):
@classmethod
def setUpTestData(cls):
cls.ml_tool = MagicMock(spec=MLTool)
cls.ml_tool.type = MLToolType.Recognizer
cls.ml_tool.slug = 'muddy'
cls.ml_tool.name = 'Muddy Recognizer'
cls.ml_tool.version = '0.42'
def test_str(self):
ds = DataSource(
type=MLToolType.Classifier,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment