Marc-André Lemburg | 3661908 | 2001-01-17 19:11:13 +0000 | [diff] [blame] | 1 | from test_support import verify, verbose, TestFailed |
Neil Schemenauer | 88c761a | 2001-07-12 13:25:53 +0000 | [diff] [blame] | 2 | import sys |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 3 | import gc |
| 4 | |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 5 | def run_test(name, thunk): |
| 6 | if verbose: |
| 7 | print "testing %s..." % name, |
| 8 | try: |
| 9 | thunk() |
| 10 | except TestFailed: |
| 11 | if verbose: |
| 12 | print "failed (expected %s but got %s)" % (result, |
| 13 | test_result) |
| 14 | raise TestFailed, name |
| 15 | else: |
| 16 | if verbose: |
| 17 | print "ok" |
| 18 | |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 19 | def test_list(): |
| 20 | l = [] |
| 21 | l.append(l) |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 22 | gc.collect() |
| 23 | del l |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 24 | if gc.collect() != 1: |
| 25 | raise TestFailed |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 26 | |
| 27 | def test_dict(): |
| 28 | d = {} |
| 29 | d[1] = d |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 30 | gc.collect() |
| 31 | del d |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 32 | if gc.collect() != 1: |
| 33 | raise TestFailed |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 34 | |
| 35 | def test_tuple(): |
Neil Schemenauer | a53cf79 | 2000-09-15 22:32:29 +0000 | [diff] [blame] | 36 | # since tuples are immutable we close the loop with a list |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 37 | l = [] |
| 38 | t = (l,) |
| 39 | l.append(t) |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 40 | gc.collect() |
| 41 | del t |
| 42 | del l |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 43 | if gc.collect() != 2: |
| 44 | raise TestFailed |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 45 | |
| 46 | def test_class(): |
| 47 | class A: |
| 48 | pass |
| 49 | A.a = A |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 50 | gc.collect() |
| 51 | del A |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 52 | if gc.collect() == 0: |
| 53 | raise TestFailed |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 54 | |
| 55 | def test_instance(): |
| 56 | class A: |
| 57 | pass |
| 58 | a = A() |
| 59 | a.a = a |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 60 | gc.collect() |
| 61 | del a |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 62 | if gc.collect() == 0: |
| 63 | raise TestFailed |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 64 | |
| 65 | def test_method(): |
Neil Schemenauer | a53cf79 | 2000-09-15 22:32:29 +0000 | [diff] [blame] | 66 | # Tricky: self.__init__ is a bound method, it references the instance. |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 67 | class A: |
| 68 | def __init__(self): |
| 69 | self.init = self.__init__ |
| 70 | a = A() |
| 71 | gc.collect() |
| 72 | del a |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 73 | if gc.collect() == 0: |
| 74 | raise TestFailed |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 75 | |
| 76 | def test_finalizer(): |
Neil Schemenauer | a53cf79 | 2000-09-15 22:32:29 +0000 | [diff] [blame] | 77 | # A() is uncollectable if it is part of a cycle, make sure it shows up |
| 78 | # in gc.garbage. |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 79 | class A: |
| 80 | def __del__(self): pass |
| 81 | class B: |
| 82 | pass |
| 83 | a = A() |
| 84 | a.a = a |
| 85 | id_a = id(a) |
| 86 | b = B() |
| 87 | b.b = b |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 88 | gc.collect() |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 89 | del a |
| 90 | del b |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 91 | if gc.collect() == 0: |
| 92 | raise TestFailed |
| 93 | for obj in gc.garbage: |
| 94 | if id(obj) == id_a: |
| 95 | del obj.a |
| 96 | break |
| 97 | else: |
| 98 | raise TestFailed |
| 99 | gc.garbage.remove(obj) |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 100 | |
| 101 | def test_function(): |
Neil Schemenauer | a53cf79 | 2000-09-15 22:32:29 +0000 | [diff] [blame] | 102 | # Tricky: f -> d -> f, code should call d.clear() after the exec to |
| 103 | # break the cycle. |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 104 | d = {} |
| 105 | exec("def f(): pass\n") in d |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 106 | gc.collect() |
| 107 | del d |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 108 | if gc.collect() != 2: |
| 109 | raise TestFailed |
| 110 | |
Neil Schemenauer | 88c761a | 2001-07-12 13:25:53 +0000 | [diff] [blame] | 111 | def test_frame(): |
| 112 | def f(): |
| 113 | frame = sys._getframe() |
| 114 | gc.collect() |
| 115 | f() |
| 116 | if gc.collect() != 1: |
| 117 | raise TestFailed |
| 118 | |
| 119 | |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 120 | def test_saveall(): |
| 121 | # Verify that cyclic garbage like lists show up in gc.garbage if the |
| 122 | # SAVEALL option is enabled. |
| 123 | debug = gc.get_debug() |
| 124 | gc.set_debug(debug | gc.DEBUG_SAVEALL) |
| 125 | l = [] |
| 126 | l.append(l) |
| 127 | id_l = id(l) |
| 128 | del l |
| 129 | gc.collect() |
| 130 | try: |
| 131 | for obj in gc.garbage: |
| 132 | if id(obj) == id_l: |
| 133 | del obj[:] |
| 134 | break |
| 135 | else: |
| 136 | raise TestFailed |
| 137 | gc.garbage.remove(obj) |
| 138 | finally: |
| 139 | gc.set_debug(debug) |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 140 | |
Neil Schemenauer | a53cf79 | 2000-09-15 22:32:29 +0000 | [diff] [blame] | 141 | def test_del(): |
| 142 | # __del__ methods can trigger collection, make this to happen |
| 143 | thresholds = gc.get_threshold() |
| 144 | gc.enable() |
| 145 | gc.set_threshold(1) |
| 146 | |
Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 147 | class A: |
| 148 | def __del__(self): |
| 149 | dir(self) |
Neil Schemenauer | a53cf79 | 2000-09-15 22:32:29 +0000 | [diff] [blame] | 150 | a = A() |
| 151 | del a |
| 152 | |
| 153 | gc.disable() |
| 154 | apply(gc.set_threshold, thresholds) |
Fred Drake | 004d5e6 | 2000-10-23 17:22:08 +0000 | [diff] [blame] | 155 | |
Jeremy Hylton | c5007aa | 2000-06-30 05:02:53 +0000 | [diff] [blame] | 156 | |
| 157 | def test_all(): |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 158 | run_test("lists", test_list) |
| 159 | run_test("dicts", test_dict) |
| 160 | run_test("tuples", test_tuple) |
| 161 | run_test("classes", test_class) |
| 162 | run_test("instances", test_instance) |
| 163 | run_test("methods", test_method) |
| 164 | run_test("functions", test_function) |
Neil Schemenauer | 88c761a | 2001-07-12 13:25:53 +0000 | [diff] [blame] | 165 | run_test("frames", test_frame) |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 166 | run_test("finalizers", test_finalizer) |
| 167 | run_test("__del__", test_del) |
| 168 | run_test("saveall", test_saveall) |
Vladimir Marangozov | f9d20c3 | 2000-08-06 22:45:31 +0000 | [diff] [blame] | 169 | |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 170 | def test(): |
| 171 | if verbose: |
| 172 | print "disabling automatic collection" |
Vladimir Marangozov | f9d20c3 | 2000-08-06 22:45:31 +0000 | [diff] [blame] | 173 | enabled = gc.isenabled() |
| 174 | gc.disable() |
Marc-André Lemburg | 3661908 | 2001-01-17 19:11:13 +0000 | [diff] [blame] | 175 | verify(not gc.isenabled() ) |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 176 | debug = gc.get_debug() |
| 177 | gc.set_debug(debug & ~gc.DEBUG_LEAK) # this test is supposed to leak |
Vladimir Marangozov | f9d20c3 | 2000-08-06 22:45:31 +0000 | [diff] [blame] | 178 | |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 179 | try: |
| 180 | test_all() |
| 181 | finally: |
| 182 | gc.set_debug(debug) |
| 183 | # test gc.enable() even if GC is disabled by default |
| 184 | if verbose: |
| 185 | print "restoring automatic collection" |
| 186 | # make sure to always test gc.enable() |
| 187 | gc.enable() |
Marc-André Lemburg | 3661908 | 2001-01-17 19:11:13 +0000 | [diff] [blame] | 188 | verify(gc.isenabled()) |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 189 | if not enabled: |
| 190 | gc.disable() |
Vladimir Marangozov | f9d20c3 | 2000-08-06 22:45:31 +0000 | [diff] [blame] | 191 | |
| 192 | |
Neil Schemenauer | faae266 | 2000-09-22 15:26:20 +0000 | [diff] [blame] | 193 | test() |