test_opcodes.py 3.61 KB
Newer Older
Guido van Rossum's avatar
Guido van Rossum committed
1 2
# Python test set -- part 2, opcodes

3
import unittest
4
from test import ann_module, support
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23

class OpcodeTest(unittest.TestCase):

    def test_try_inside_for_loop(self):
        n = 0
        for i in range(10):
            n = n+i
            try: 1/0
            except NameError: pass
            except ZeroDivisionError: pass
            except TypeError: pass
            try: pass
            except: pass
            try: pass
            finally: pass
            n = n+i
        if n != 90:
            self.fail('try inside for')

24 25 26 27 28 29
    def test_setup_annotations_line(self):
        # check that SETUP_ANNOTATIONS does not create spurious line numbers
        try:
            with open(ann_module.__file__) as f:
                txt = f.read()
            co = compile(txt, ann_module.__file__, 'exec')
30
            self.assertEqual(co.co_firstlineno, 3)
31 32 33 34 35 36 37 38 39 40 41 42 43 44
        except OSError:
            pass

    def test_no_annotations_if_not_needed(self):
        class C: pass
        with self.assertRaises(AttributeError):
            C.__annotations__

    def test_use_existing_annotations(self):
        ns = {'__annotations__': {1: 2}}
        exec('x: int', ns)
        self.assertEqual(ns['__annotations__'], {'x': int, 1: 2})

    def test_do_not_recreate_annotations(self):
45
        # Don't rely on the existence of the '__annotations__' global.
46 47
        with support.swap_item(globals(), '__annotations__', {}):
            del globals()['__annotations__']
48 49
            class C:
                del __annotations__
50 51
                with self.assertRaises(NameError):
                    x: int
52

53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77
    def test_raise_class_exceptions(self):

        class AClass(Exception): pass
        class BClass(AClass): pass
        class CClass(Exception): pass
        class DClass(AClass):
            def __init__(self, ignore):
                pass

        try: raise AClass()
        except: pass

        try: raise AClass()
        except AClass: pass

        try: raise BClass()
        except AClass: pass

        try: raise BClass()
        except CClass: self.fail()
        except: pass

        a = AClass()
        b = BClass()

78 79
        try:
            raise b
80
        except AClass as v:
81
            self.assertEqual(v, b)
82 83 84 85 86 87 88 89
        else:
            self.fail("no exception")

        # not enough arguments
        ##try:  raise BClass, a
        ##except TypeError: pass
        ##else: self.fail("no exception")

90
        try:  raise DClass(a)
91
        except DClass as v:
92
            self.assertIsInstance(v, DClass)
93 94 95 96 97 98 99
        else:
            self.fail("no exception")

    def test_compare_function_objects(self):

        f = eval('lambda: None')
        g = eval('lambda: None')
100
        self.assertNotEqual(f, g)
101 102 103

        f = eval('lambda a: a')
        g = eval('lambda a: a')
104
        self.assertNotEqual(f, g)
105 106 107

        f = eval('lambda a=1: a')
        g = eval('lambda a=1: a')
108
        self.assertNotEqual(f, g)
109 110 111

        f = eval('lambda: 0')
        g = eval('lambda: 1')
112
        self.assertNotEqual(f, g)
113 114 115

        f = eval('lambda: None')
        g = eval('lambda a: None')
116
        self.assertNotEqual(f, g)
117 118 119

        f = eval('lambda a: None')
        g = eval('lambda b: None')
120
        self.assertNotEqual(f, g)
121 122 123

        f = eval('lambda a: None')
        g = eval('lambda a=None: None')
124
        self.assertNotEqual(f, g)
125 126 127

        f = eval('lambda a=0: None')
        g = eval('lambda a=1: None')
128
        self.assertNotEqual(f, g)
129

130 131 132 133 134 135
    def test_modulo_of_string_subclasses(self):
        class MyString(str):
            def __mod__(self, value):
                return 42
        self.assertEqual(MyString() % 3, 42)

136 137

if __name__ == '__main__':
138
    unittest.main()