def test_Composite(self):
'Test Composite'
expr = pyson.If(pyson.Not(
pyson.In('company', pyson.Eval('context', {}))), '=', '!=')
eval = pyson.PYSONEncoder().encode(['id', expr,
pyson.Get(pyson.Eval('context', {}), 'company', -1)])
self.assertEqual(pyson.PYSONDecoder({'context': {'company': 1}}
).decode(eval), ['id', '!=', 1])
self.assertEqual(pyson.PYSONDecoder({'context': {}}
).decode(eval), ['id', '=', -1])
self.assertEqual(repr(expr),
"If(Not(In('company', Eval('context', {}))), '=', '!=')")
def test_noeval(self):
decoder = pyson.PYSONDecoder(noeval=True)
encoder = pyson.PYSONEncoder()
for instance in [
pyson.Eval('test', 0),
pyson.Not(True),
pyson.Bool('test'),
pyson.And(True, False, True),
pyson.Or(False, True, True),
pyson.Equal('foo', 'bar'),
pyson.Greater(1, 0),
pyson.Less(0, 1),
pyson.If(True, 'foo', 'bar'),
pyson.Get({'foo': 'bar'}, 'foo', 'default'),
pyson.In('foo', ['foo', 'bar']),
pyson.Date(),
pyson.DateTime(),
pyson.Len([1, 2, 3]),
]:
self.assertEqual(decoder.decode(encoder.encode(instance)).pyson(),
instance.pyson())
def test_In(self):
'Test pyson.In'
self.assertEqual(
pyson.In('foo', {
'foo': 'bar'
}).pyson(), {
'__class__': 'In',
'k': 'foo',
'v': {
'foo': 'bar'
},
})
self.assertRaises(AssertionError, pyson.In, object(), {})
self.assertRaises(AssertionError, pyson.In, 'test', 'foo')
self.assertEqual(pyson.In('foo', {}).types(), set([bool]))
eval = pyson.PYSONEncoder().encode(pyson.In('foo', {'foo': 'bar'}))
self.assertTrue(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('1', {1: 'bar'}))
self.assertTrue(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('test', {'foo': 'bar'}))
self.assertFalse(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('1', {2: 'bar'}))
self.assertFalse(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('test', {}))
self.assertFalse(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('foo', ['foo']))
self.assertTrue(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In(1, [1]))
self.assertTrue(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('test', ['foo']))
self.assertFalse(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In(1, [2]))
self.assertFalse(pyson.PYSONDecoder().decode(eval))
eval = pyson.PYSONEncoder().encode(pyson.In('test', []))
self.assertFalse(pyson.PYSONDecoder().decode(eval))
self.assertEqual(repr(pyson.In('foo', ['foo', 'bar'])),
"In('foo', ['foo', 'bar'])")
