test_cli.py 560 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import logging
from functools import partial

logger = logging.getLogger(__name__)

from unittest import TestCase
from senpy.cli import main_function
from senpy.models import Error


class CLITest(TestCase):
    def test_basic(self):
        self.assertRaises(Error, partial(main_function, []))
        res = main_function(['--input', 'test'])
        assert 'entries' in res
        res = main_function(['--input', 'test', '--algo', 'rand'])
        assert 'entries' in res
        assert 'analysis' in res
        assert res['analysis'][0]['name'] == 'rand'