def test_If(self): 'Test pyson.If' self.assertEqual( pyson.If(True, 'foo', 'bar').pyson(), { '__class__': 'If', 'c': True, 't': 'foo', 'e': 'bar', }) self.assertRaises(AssertionError, pyson.If, 'test', 'foo', 'bar') self.assertRaises(AssertionError, pyson.If, True, 'foo', False) self.assertEqual( pyson.If(True, 'foo', 'bar').types(), set([basestring])) self.assertEqual(pyson.If(True, False, True).types(), set([bool])) eval = pyson.PYSONEncoder().encode(pyson.If(True, 'foo', 'bar')) self.assertEqual(pyson.PYSONDecoder().decode(eval), 'foo') eval = pyson.PYSONEncoder().encode(pyson.If(False, 'foo', 'bar')) self.assertEqual(pyson.PYSONDecoder().decode(eval), 'bar') self.assertEqual(repr(pyson.If(True, 'foo', 'bar')), "If(True, 'foo', 'bar')")
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())