Exemplos de set_args em Python

Exemplo n.º 1

Arquivo: test_util.py Projeto: shibli049/markovchain

    def testSettings(self):
        args = Namespace()
        set_args(args, ())
        self.assertIsNone(args.settings)

        args = Namespace(settings=None)
        set_args(args, ())
        self.assertEqual(args.settings, {})

        s = list(range(3))
        mock = MagicMock(return_value=s)
        args = Namespace(settings=mock)
        set_args(args, ())
        self.assertEqual(args.settings, s)
        mock.close.assert_called_with()

Exemplo n.º 2

def test_set_args_settings(mocker):
    mocker.patch('json.load', new=lambda x: x())

    args = Namespace()
    set_args(args)
    assert args.settings == {}  # pylint:disable=no-member

    args = Namespace(settings=None)
    set_args(args)
    assert args.settings == {}  # pylint:disable=no-member

    s = list(range(3))
    mock = MagicMock(return_value=s)
    args = Namespace(settings=mock)
    set_args(args)
    assert args.settings == s  # pylint:disable=no-member
    mock.close.assert_called_with()

Exemplo n.º 3

Arquivo: test_util.py Projeto: shibli049/markovchain

    def testType(self):
        args = Namespace()
        set_args(args, ())
        self.assertEqual(args.type, JSON)
        self.assertTrue(issubclass(args.markov, MarkovBase))
        self.assertTrue(issubclass(args.markov, MarkovJsonMixin))

        args = Namespace()
        set_args(args, (Mock,))
        self.assertEqual(args.type, JSON)
        self.assertTrue(issubclass(args.markov, Mock))
        self.assertTrue(issubclass(args.markov, MarkovBase))
        self.assertTrue(issubclass(args.markov, MarkovJsonMixin))

        args = Namespace(output='.db')
        set_args(args, ())
        self.assertEqual(args.type, SQLITE)
        self.assertTrue(issubclass(args.markov, MarkovBase))
        self.assertTrue(issubclass(args.markov, MarkovSqliteMixin))

        args = Namespace(output='.db', state='.json.bz2')
        set_args(args, ())
        self.assertEqual(args.type, JSON)

        args = Namespace(output='.db', state='.json.bz2', type='sqlite')
        set_args(args, ())
        self.assertEqual(args.type, SQLITE)

        args = Namespace(output='.db', state='.json.bz2', type='json')
        set_args(args, ())
        self.assertEqual(args.type, JSON)

Exemplo n.º 4

Arquivo: test_util.py Projeto: shibli049/markovchain

 def testErrors(self):
     args = Namespace(output=sys.stdout, progress=True)
     with self.assertRaises(ValueError):
         set_args(args, ())

Exemplo n.º 5

def test_set_args_type(args, res):
    set_args(args)
    assert args.type == res

Exemplo n.º 6

def test_set_args_error():
    args = Namespace(output=sys.stdout, progress=True)
    with pytest.raises(ValueError):
        set_args(args)