cmlenz@1: # -*- coding: utf-8 -*- cmlenz@1: # cmlenz@1: # Copyright (C) 2006 Christopher Lenz cmlenz@1: # All rights reserved. cmlenz@1: # cmlenz@1: # This software is licensed as described in the file COPYING, which cmlenz@1: # you should have received as part of this distribution. The terms cmlenz@27: # are also available at http://markup.cmlenz.net/wiki/License. cmlenz@1: # cmlenz@1: # This software consists of voluntary contributions made by many cmlenz@1: # individuals. For the exact contribution history, see the revision cmlenz@27: # history and logs, available at hhttp://markup.cmlenz.net/log/. cmlenz@1: cmlenz@1: import doctest cmlenz@1: import unittest cmlenz@1: cmlenz@27: from markup.eval import Expression cmlenz@1: cmlenz@30: cmlenz@30: class ExpressionTestCase(unittest.TestCase): cmlenz@30: cmlenz@30: def test_str_literal(self): cmlenz@30: self.assertEqual('foo', Expression('"foo"').evaluate({})) cmlenz@30: self.assertEqual('foo', Expression('"""foo"""').evaluate({})) cmlenz@30: self.assertEqual('foo', Expression("'foo'").evaluate({})) cmlenz@30: self.assertEqual('foo', Expression("'''foo'''").evaluate({})) cmlenz@30: self.assertEqual('foo', Expression("u'foo'").evaluate({})) cmlenz@30: self.assertEqual('foo', Expression("r'foo'").evaluate({})) cmlenz@30: cmlenz@30: def test_num_literal(self): cmlenz@30: self.assertEqual(42, Expression("42").evaluate({})) cmlenz@30: self.assertEqual(42L, Expression("42L").evaluate({})) cmlenz@30: self.assertEqual(.42, Expression(".42").evaluate({})) cmlenz@30: self.assertEqual(07, Expression("07").evaluate({})) cmlenz@30: self.assertEqual(0xF2, Expression("0xF2").evaluate({})) cmlenz@30: self.assertEqual(0XF2, Expression("0XF2").evaluate({})) cmlenz@30: cmlenz@30: def test_dict_literal(self): cmlenz@30: self.assertEqual({}, Expression("{}").evaluate({})) cmlenz@30: self.assertEqual({'key': True}, cmlenz@30: Expression("{'key': value}").evaluate({'value': True})) cmlenz@30: cmlenz@30: def test_list_literal(self): cmlenz@30: self.assertEqual([], Expression("[]").evaluate({})) cmlenz@30: self.assertEqual([1, 2, 3], Expression("[1, 2, 3]").evaluate({})) cmlenz@30: self.assertEqual([True], cmlenz@30: Expression("[value]").evaluate({'value': True})) cmlenz@30: cmlenz@30: def test_tuple_literal(self): cmlenz@30: self.assertEqual((), Expression("()").evaluate({})) cmlenz@30: self.assertEqual((1, 2, 3), Expression("(1, 2, 3)").evaluate({})) cmlenz@30: self.assertEqual((True,), cmlenz@30: Expression("(value,)").evaluate({'value': True})) cmlenz@30: cmlenz@31: def test_unaryop_pos(self): cmlenz@31: self.assertEqual(1, Expression("+1").evaluate({})) cmlenz@31: self.assertEqual(1, Expression("+x").evaluate({'x': 1})) cmlenz@31: cmlenz@31: def test_unaryop_neg(self): cmlenz@31: self.assertEqual(-1, Expression("-1").evaluate({})) cmlenz@31: self.assertEqual(-1, Expression("-x").evaluate({'x': 1})) cmlenz@31: cmlenz@31: def test_unaryop_not(self): cmlenz@31: self.assertEqual(False, Expression("not True").evaluate({})) cmlenz@31: self.assertEqual(False, Expression("not x").evaluate({'x': True})) cmlenz@31: cmlenz@31: def test_binop_add(self): cmlenz@31: self.assertEqual(3, Expression("2 + 1").evaluate({})) cmlenz@31: self.assertEqual(3, Expression("x + y").evaluate({'x': 2, 'y': 1})) cmlenz@31: cmlenz@31: def test_binop_sub(self): cmlenz@31: self.assertEqual(1, Expression("2 - 1").evaluate({})) cmlenz@31: self.assertEqual(1, Expression("x - y").evaluate({'x': 1, 'y': 1})) cmlenz@31: cmlenz@31: def test_binop_sub(self): cmlenz@31: self.assertEqual(1, Expression("2 - 1").evaluate({})) cmlenz@31: self.assertEqual(1, Expression("x - y").evaluate({'x': 2, 'y': 1})) cmlenz@31: cmlenz@31: def test_binop_mul(self): cmlenz@31: self.assertEqual(4, Expression("2 * 2").evaluate({})) cmlenz@31: self.assertEqual(4, Expression("x * y").evaluate({'x': 2, 'y': 2})) cmlenz@31: cmlenz@31: def test_binop_pow(self): cmlenz@31: self.assertEqual(4, Expression("2 ** 2").evaluate({})) cmlenz@31: self.assertEqual(4, Expression("x ** y").evaluate({'x': 2, 'y': 2})) cmlenz@31: cmlenz@31: def test_binop_div(self): cmlenz@31: self.assertEqual(2, Expression("4 / 2").evaluate({})) cmlenz@31: self.assertEqual(2, Expression("x / y").evaluate({'x': 4, 'y': 2})) cmlenz@31: cmlenz@31: def test_binop_floordiv(self): cmlenz@31: self.assertEqual(1, Expression("3 // 2").evaluate({})) cmlenz@31: self.assertEqual(1, Expression("x // y").evaluate({'x': 3, 'y': 2})) cmlenz@31: cmlenz@31: def test_binop_mod(self): cmlenz@31: self.assertEqual(1, Expression("3 % 2").evaluate({})) cmlenz@31: self.assertEqual(1, Expression("x % y").evaluate({'x': 3, 'y': 2})) cmlenz@31: cmlenz@31: def test_binop_contains(self): cmlenz@31: self.assertEqual(True, Expression("1 in (1, 2, 3)").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x in y").evaluate({'x': 1, cmlenz@31: 'y': (1, 2, 3)})) cmlenz@31: cmlenz@31: def test_binop_not_contains(self): cmlenz@31: self.assertEqual(True, Expression("4 not in (1, 2, 3)").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x not in y").evaluate({'x': 4, cmlenz@31: 'y': (1, 2, 3)})) cmlenz@31: cmlenz@31: def test_boolop_and(self): cmlenz@31: self.assertEqual(False, Expression("True and False").evaluate({})) cmlenz@31: self.assertEqual(False, Expression("x and y").evaluate({'x': True, cmlenz@31: 'y': False})) cmlenz@31: cmlenz@31: def test_boolop_or(self): cmlenz@31: self.assertEqual(True, Expression("True or False").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x or y").evaluate({'x': True, cmlenz@31: 'y': False})) cmlenz@31: cmlenz@31: def test_compare_eq(self): cmlenz@31: self.assertEqual(True, Expression("1 == 1").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x == y").evaluate({'x': 1, 'y': 1})) cmlenz@31: cmlenz@31: def test_compare_ne(self): cmlenz@31: self.assertEqual(False, Expression("1 != 1").evaluate({})) cmlenz@31: self.assertEqual(False, Expression("x != y").evaluate({'x': 1, 'y': 1})) cmlenz@31: cmlenz@31: def test_compare_lt(self): cmlenz@31: self.assertEqual(True, Expression("1 < 2").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x < y").evaluate({'x': 1, 'y': 2})) cmlenz@31: cmlenz@31: def test_compare_le(self): cmlenz@31: self.assertEqual(True, Expression("1 <= 1").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x <= y").evaluate({'x': 1, 'y': 1})) cmlenz@31: cmlenz@31: def test_compare_gt(self): cmlenz@31: self.assertEqual(True, Expression("2 > 1").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x > y").evaluate({'x': 2, 'y': 1})) cmlenz@31: cmlenz@31: def test_compare_ge(self): cmlenz@31: self.assertEqual(True, Expression("1 >= 1").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x >= y").evaluate({'x': 1, 'y': 1})) cmlenz@31: cmlenz@31: def test_compare_multi(self): cmlenz@31: self.assertEqual(True, Expression("1 != 3 == 3").evaluate({})) cmlenz@31: self.assertEqual(True, Expression("x != y == y").evaluate({'x': 3, cmlenz@31: 'y': 3})) cmlenz@31: cmlenz@30: cmlenz@1: def suite(): cmlenz@1: suite = unittest.TestSuite() cmlenz@30: suite.addTest(unittest.makeSuite(ExpressionTestCase, 'test')) cmlenz@27: suite.addTest(doctest.DocTestSuite(Expression.__module__)) cmlenz@1: return suite cmlenz@1: cmlenz@1: if __name__ == '__main__': cmlenz@1: unittest.main(defaultTest='suite')