test_validate.py 文件源码

def test_get_mongo(self):
        "Test get_mongo() function"
        import math
        with self.assertRaises(ValueError):
            for rec in [1, 2, 3], 'foo':
                vv.mongo_get(rec, 'a')
        for rec in {}, None, '':
            self.assertEqual(vv.mongo_get(rec, 'a', 'dee-fault'), 'dee-fault')
        rec = {'a': {
                'apple': {
                    'red': 'delicious',
                    'green': 'grannysmith',
                    'blue': 'venutian'},
                'anumber': 2, },
                'e': math.e, }
        default = 'mittens'
        # these should be found
        for key, expected in (('a.apple.green', 'grannysmith'), ('a.anumber', 2), ('e', math.e)):
            self.assertEqual(vv.mongo_get(rec, key, default), expected)
            # these should not
        for key in 'a.apple.orange', 'x', '', 'z.zebra.', '.':
            self.assertEqual(vv.mongo_get(rec, key, default), default)