1edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep"""Test compiler changes for unary ops (+, -, ~) introduced in Python 2.2"""
2edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
3edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoepimport unittest
4edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoepfrom test.test_support import run_unittest, have_unicode
5edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
6edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoepclass UnaryOpTestCase(unittest.TestCase):
7edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
8edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    def test_negative(self):
9edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-2 == 0 - 2)
10edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-0 == 0)
11edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(--2 == 2)
12edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-2L == 0 - 2L)
13edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-2.0 == 0 - 2.0)
14edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-2j == 0 - 2j)
15edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
16edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    def test_positive(self):
17edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(+2 == 2)
18edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(+0 == 0)
19edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(++2 == 2)
20edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(+2L == 2L)
21edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(+2.0 == 2.0)
22edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(+2j == 2j)
23edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
24edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    def test_invert(self):
25edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-2 == 0 - 2)
26edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-0 == 0)
27edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(--2 == 2)
28edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(-2L == 0 - 2L)
29edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
30edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    def test_no_overflow(self):
31edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        nines = "9" * 32
32edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(eval("+" + nines) == eval("+" + nines + "L"))
33edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(eval("-" + nines) == eval("-" + nines + "L"))
34edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertTrue(eval("~" + nines) == eval("~" + nines + "L"))
35edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
36edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    def test_negation_of_exponentiation(self):
37edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        # Make sure '**' does the right thing; these form a
38edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        # regression test for SourceForge bug #456756.
39edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertEqual(-2 ** 3, -8)
40edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertEqual((-2) ** 3, -8)
41edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertEqual(-2 ** 4, -16)
42edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertEqual((-2) ** 4, 16)
43edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
44edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    def test_bad_types(self):
45edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        for op in '+', '-', '~':
46edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep            self.assertRaises(TypeError, eval, op + "'a'")
47edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep            if have_unicode:
48edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep                self.assertRaises(TypeError, eval, op + "u'a'")
49edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
50edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertRaises(TypeError, eval, "~2j")
51edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep        self.assertRaises(TypeError, eval, "~2.0")
52edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
53edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
54edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoepdef test_main():
55edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    run_unittest(UnaryOpTestCase)
56edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
57edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep
58edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoepif __name__ == "__main__":
59edbb763a2b63074cd468a5d33a17908b2cc0654Jeff Vander Stoep    test_main()
60