From 77f99ac0c0717ee20b701d5c20fea0f00ed4bc40 Mon Sep 17 00:00:00 2001
From: Johan Dahlin <johan@src.gnome.org>
Date: Sat, 20 Jan 2007 14:30:45 +0000
Subject: clean up gc collection

svn path=/trunk/; revision=2807
---
 tests/test_gdk.py | 30 ++++++++++++++----------------
 1 file changed, 14 insertions(+), 16 deletions(-)

(limited to 'tests')

diff --git a/tests/test_gdk.py b/tests/test_gdk.py
index 77fe5cc1..e23c4583 100644
--- a/tests/test_gdk.py
+++ b/tests/test_gdk.py
@@ -47,6 +47,15 @@ class GdkTest(unittest.TestCase):
                                     1,
                                     [0xdeadbe, 0xbebabe])
 
+    def _collect(self):
+        cnt = 0
+        while True:
+            x = gc.collect()
+            cnt += x
+            if x:
+                break
+        return cnt
+
     def testDisplay(self):
         while gc.collect():
             pass
@@ -54,22 +63,11 @@ class GdkTest(unittest.TestCase):
         display = gtk.gdk.Display(None)
         del display
 
-        cnt = 0
-        while True:
-            x = gc.collect()
-            cnt += x
-            if x:
-                break
-        self.assertEquals(cnt, 1)
-        
+        self.assertEquals(self._collect(), 1)
+
         display = gtk.gdk.Display(None)
+        self.assertEquals(display.__grefcount__, 1)
         display.close()
+        self.assertEquals(display.__grefcount__, 1)
         del display
-
-        cnt = 0
-        while True:
-            x = gc.collect()
-            cnt += x
-            if x:
-                break
-        self.assertEquals(cnt, 1)
+        self.assertEquals(self._collect(), 1)
-- 
cgit v1.2.1