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()
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()
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)
def testErrors(self): args = Namespace(output=sys.stdout, progress=True) with self.assertRaises(ValueError): set_args(args, ())
def test_set_args_type(args, res): set_args(args) assert args.type == res
def test_set_args_error(): args = Namespace(output=sys.stdout, progress=True) with pytest.raises(ValueError): set_args(args)