Skip to content
Snippets Groups Projects
Commit 0d13d154 authored by Bastien Abadie's avatar Bastien Abadie Committed by Erwan Rouchet
Browse files

Revert "Remove MLTool unique unit test."

This reverts commit 5927a444.
parent cb83de03
No related branches found
No related tags found
No related merge requests found
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