From 131c6a25e900029271fca82d454b2dee0deab255 Mon Sep 17 00:00:00 2001 From: Daniele Varrazzo Date: Sun, 2 Jan 2011 17:09:30 +0100 Subject: [PATCH] Unregister test adapters to keep a more precise references count --- tests/types_basic.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tests/types_basic.py b/tests/types_basic.py index 5bcff062..c7e639e9 100755 --- a/tests/types_basic.py +++ b/tests/types_basic.py @@ -166,7 +166,11 @@ class AdaptSubclassTest(unittest.TestCase): register_adapter(A, lambda a: AsIs("a")) register_adapter(B, lambda b: AsIs("b")) - self.assertEqual('b', adapt(C()).getquoted()) + try: + self.assertEqual('b', adapt(C()).getquoted()) + finally: + del psycopg2.extensions.adapters[A, psycopg2.extensions.ISQLQuote] + del psycopg2.extensions.adapters[B, psycopg2.extensions.ISQLQuote] def test_no_mro_no_joy(self): from psycopg2.extensions import adapt, register_adapter, AsIs @@ -175,7 +179,11 @@ class AdaptSubclassTest(unittest.TestCase): class B(A): pass register_adapter(A, lambda a: AsIs("a")) - self.assertRaises(psycopg2.ProgrammingError, adapt, B()) + try: + self.assertRaises(psycopg2.ProgrammingError, adapt, B()) + finally: + del psycopg2.extensions.adapters[A, psycopg2.extensions.ISQLQuote] + def test_suite(): return unittest.TestLoader().loadTestsFromName(__name__)