[pypy-commit] pypy stm-gc: Improve the tests.

[arigo]

Author: Armin Rigo <arigo at tunes.org>
Branch: stm-gc
Changeset: r54722:9cb2b55a043e
Date: 2012-04-24 15:50 +0200
http://bitbucket.org/pypy/pypy/changeset/9cb2b55a043e/

Log:	Improve the tests.

diff --git a/pypy/rpython/memory/gc/test/test_stmgc.py b/pypy/rpython/memory/gc/test/test_stmgc.py
--- a/pypy/rpython/memory/gc/test/test_stmgc.py
+++ b/pypy/rpython/memory/gc/test/test_stmgc.py
@@ -783,11 +783,15 @@
         sr1, sr1_adr = self.malloc(SR, globl=False)
         wr1, wr1_adr = self.malloc(WR, globl=False, weakref=True)
         wr1.wadr = sr1_adr
+        #
         self.gc.root_walker.current_stack = [wr1]
         self.gc.collect(0)
-        #
         [wr1] = self.gc.root_walker.current_stack
         assert not wr1.wadr        # weakref to dead object
+        #
+        self.gc.collect(0)
+        assert self.gc.root_walker.current_stack == [wr1]
+        assert not wr1.wadr
 
     def test_weakref_to_global_turned_local(self):
         from pypy.rpython.memory.gc.test import test_stmtls
@@ -803,10 +807,15 @@
         wr2, wr2_adr = self.malloc(WR, globl=False, weakref=True)
         wr1.wadr = sr1_adr
         wr2.wadr = sr2_adr
+        #
         self.gc.root_walker.current_stack = [wr1, wr2]
         self.gc.collect(0)
+        [wr1, wr2] = self.gc.root_walker.current_stack
+        assert wr1.wadr == sr1_adr
+        assert wr2.wadr == sr2_adr
         #
-        [wr1, wr2] = self.gc.root_walker.current_stack
+        self.gc.collect(0)
+        assert self.gc.root_walker.current_stack == [wr1, wr2]
         assert wr1.wadr == sr1_adr
         assert wr2.wadr == sr2_adr