Beispiel #1
0
	def test_optional(self):
		t = typed.dict({'a': typed.int, 'b': typed.bool.optional, 'c': typed.list(typed.string).optional})

		self.assertTrue(t.test({'a': 1, 'b': True, 'c': ['X', 'Y']}))
		self.assertTrue(t.test({'a': 1}))
		self.assertTrue(t.test({'a': 1, 'b': True}))
		self.assertTrue(t.test({'a': 1, 'c': ['X', 'Y']}))

		self.assertFalse(t.test({}))
		self.assertFalse(t.test({'a': 1, 'b': 0.01}))
		self.assertFalse(t.test({'a': 1, 'b': True, 'c': 'foo'}))
		self.assertFalse(t.test({'a': 1, 'b': True, 'c': [1, 2, 3]}))
		self.assertFalse(t.test({'a': 1, 'b': True, 'c': ['X', 'Y'], 'd': 'bar'}))
Beispiel #2
0
	def test_list(self):
		t1 = typed.list(typed.int)

		self.assertEqual(t1.load([1, 2, 3]), [1, 2, 3])
		self.assertEqual(t1.save([1, 2, 3]), [1, 2, 3])

		self.assertRaises(ValueError, t1.load, [1, '2', None])
		self.assertRaises(ValueError, t1.save, [1, '2', None])

		for value in [datetime.datetime.now(), 1, 'a', True, [1.2, 2.3, 3.4], 1.2, None, {'a': 1}, self]:
			self.assertRaises(ValueError, t1.load, value)
			self.assertRaises(ValueError, t1.save, value)

		t2 = typed.list(typed.int | typed.string)

		self.assertEqual(t2.load([1, 'a', 2, 'b']), [1, 'a', 2, 'b'])
		self.assertEqual(t2.save([1, 'a', 2, 'b']), [1, 'a', 2, 'b'])

		t3 = typed.list(typed.int) | typed.list(typed.string)

		self.assertRaises(ValueError, t3.load, [1, 'a', 2, 'b'])
		self.assertRaises(ValueError, t3.save, [1, 'a', 2, 'b'])

		self.assertEqual(t3.load([1, 2, 3]), [1, 2, 3])
		self.assertEqual(t3.save([1, 2, 3]), [1, 2, 3])
		self.assertEqual(t3.load(['a', 'b', 'c']), ['a', 'b', 'c'])
		self.assertEqual(t3.save(['a', 'b', 'c']), ['a', 'b', 'c'])

		datetime_format = '%Y-%m-%d %H:%M:%S'
		dt_list = [datetime.datetime(2013, 9, 21, 11, 42, 33), datetime.datetime(2012, 3, 24, 9, 6, 13)]
		dt_str_list = map(lambda dt: dt.strftime(datetime_format), dt_list)

		t4 = typed.list(typed.datetime.format(datetime_format))

		self.assertEqual(t4.load(list(dt_str_list)), dt_list)
		self.assertEqual(t4.save(list(dt_list)), dt_str_list)
Beispiel #3
0
	def test_dict(self):
		self.assertRaises(TypeError, typed.dict, 1)
		self.assertRaises(TypeError, typed.dict, {'a', 1})
		self.assertRaises(TypeError, typed.dict, {'a', int})

		t = typed.dict({'a': typed.int, 'b': typed.bool, 'c': typed.list(typed.string)})

		self.assertTrue(t.test({'a': 1, 'b': True, 'c': ['X', 'Y']}))

		self.assertFalse(t.test({}))
		self.assertFalse(t.test({'a': 1}))
		self.assertFalse(t.test({'a': 1, 'b': True}))
		self.assertFalse(t.test({'a': 1, 'b': True, 'c': 'foo'}))
		self.assertFalse(t.test({'a': 1, 'b': True, 'c': [1, 2, 3]}))
		self.assertFalse(t.test({'a': 1, 'b': True, 'c': ['X', 'Y'], 'd': 'bar'}))
Beispiel #4
0
	def test_list(self):
		self.assertRaises(TypeError, typed.list, int)

		t1 = typed.list(typed.int)
		self.assertTrue(isinstance(t1, typed.ListType))

		self.assertTrue(t1.test([1, 2, 3]))
		self.assertTrue(t1.test([]))

		self.assertFalse(t1.test([1, '2', None]))
		self.assertFalse(t1.test(tuple([1, 2, 3])))

		for value in [datetime.datetime.now(), 1, 'a', True, [1.2, 2.3, 3.4], 1.2, None, {'a': 1}, self]:
			self.assertFalse(t1.test(value))

		t2 = typed.list(typed.int | typed.string)

		self.assertTrue(t2.test([1, 'a', 2, 'b']))

		t3 = typed.list(typed.int) | typed.list(typed.string)

		self.assertFalse(t3.test([1, 'a', 2, 'b']))
		self.assertTrue(t3.test([1, 2, 3]))
		self.assertTrue(t3.test(['a', 'b', 'c']))
Beispiel #5
0
	def test_format_json(self):
		t1 = typed.dict({
				'a': typed.datetime.format('%Y-%m-%d %H:%M:%S').optional,
				'b': (typed.string | typed.none).optional,
				'c': typed.bool.format({True: 1, False: 0}).default(False),
			})

		t2 = typed.dict({
				'd': typed.int,
				'e': (typed.none | typed.date.format('%d/%m/%Y')).default(None),
				'f': typed.list(typed.int),
				'g': typed.json(t1).default(t1.load({})),
			}).format(typed.json)

		self.assertEqual(t2.load(json.dumps({'d': 1, 'f': []})), {'d': 1, 'e': None, 'f': [], 'g': {'c': False}})
		self.assertEqual(t2.load(json.dumps({'d': 0, 'e': None, 'f': [1, 2], 'g': json.dumps({'b': None, 'c': 1})})), {'d': 0, 'e': None, 'f': [1, 2], 'g': {'b': None, 'c': True}})
		self.assertEqual(t2.load(json.dumps({'d': 1, 'e': '15/02/2011', 'f': [], 'g': json.dumps({'a': '2013-11-11 16:16:16', 'b': 'foo', 'c': 0})})), {'d': 1, 'e': datetime.date(2011, 02, 15), 'f': [], 'g': {'a': datetime.datetime(2013, 11, 11, 16, 16, 16), 'b': 'foo', 'c': False}})
Beispiel #6
0
			}).format(typed.json)

		self.assertEqual(t2.load(json.dumps({'d': 1, 'f': []})), {'d': 1, 'e': None, 'f': [], 'g': {'c': False}})
		self.assertEqual(t2.load(json.dumps({'d': 0, 'e': None, 'f': [1, 2], 'g': json.dumps({'b': None, 'c': 1})})), {'d': 0, 'e': None, 'f': [1, 2], 'g': {'b': None, 'c': True}})
		self.assertEqual(t2.load(json.dumps({'d': 1, 'e': '15/02/2011', 'f': [], 'g': json.dumps({'a': '2013-11-11 16:16:16', 'b': 'foo', 'c': 0})})), {'d': 1, 'e': datetime.date(2011, 02, 15), 'f': [], 'g': {'a': datetime.datetime(2013, 11, 11, 16, 16, 16), 'b': 'foo', 'c': False}})

		self.assertEqual(t2.save({'d': 1, 'f': [1, 2, 3, 4]}), json.dumps({'d': 1, 'f': [1, 2, 3, 4]}))
		self.assertEqual(t2.save({'d': 1, 'e': None, 'f': [1, 2, 3, 4], 'g': {'c': False}}), json.dumps({'d': 1, 'f': [1, 2, 3, 4]}))
		self.assertEqual(t2.save({'d': 1, 'e': datetime.date(2013, 02, 15), 'f': [], 'g': {'a': datetime.datetime(2013, 10, 10, 05, 55, 50), 'b': None, 'c': True}}), json.dumps({'d': 1, 'e': '15/02/2013', 'f': [], 'g': json.dumps({'a': '2013-10-10 05:55:50', 'b': None, 'c': 1})}))

		if HAS_UJSON:

			t3 = typed.json(typed.dict({
					'a': typed.int,
					'b': typed.float,
					'c': typed.list(typed.float),
					'd': typed.dict({
							'e': typed.list(typed.float),
						}),
				}), double_precision=4)

			obj = {'a': 1, 'b': 0.0123456789, 'c': [0.00001, 0.00011, 0.00026], 'd': {'e': [10000.098765432, 123456.01001]}}
			obj_json = '{"a": 1, "b": 0.0123, "c": [0.0, 0.0001, 0.0003], "d": {"e": [10000.0988, 123456.01]}}'

			self.assertItemsEqual(t3.save(obj)[2:-1].split(',"'), obj_json.replace(' ', '')[2:-1].split(',"'))		# elements might come in different order


	def test_tuple(self):
		t1 = typed.tuple(typed.int, typed.datetime.format('%Y-%m-%d %H:%M:%S'), typed.string | typed.none)

		self.assertEqual(t1.load(tuple([4, '2013-10-11 11:02:45', 'foo'])), (4, datetime.datetime(2013, 10, 11, 11, 02, 45), 'foo'))