Skip to content
Snippets Groups Projects
Commit 5927a444 authored by Bastien Abadie's avatar Bastien Abadie
Browse files

Remove MLTool unique unit test.

parent cb83de03
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.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,
slug='something',
name='some classifier',
revision='1.2.3',
internal=False,
)
self.assertEqual(str(ds), 'something 1.2.3')
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