mirror of https://github.com/gsi-upm/senpy
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
740 B
Python
28 lines
740 B
Python
import logging
|
|
from functools import partial
|
|
|
|
try:
|
|
from unittest.mock import patch
|
|
except ImportError:
|
|
from mock import patch
|
|
|
|
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, []))
|
|
|
|
with patch('senpy.extensions.Senpy.analyse') as patched:
|
|
main_function(['--input', 'test'])
|
|
|
|
patched.assert_called_with(input='test')
|
|
with patch('senpy.extensions.Senpy.analyse') as patched:
|
|
main_function(['--input', 'test', '--algo', 'rand'])
|
|
|
|
patched.assert_called_with(input='test', algo='rand')
|