- fixed clear_mappers() behavior to better clean up after itself

1 files changed, 22 insertions, 0 deletions

diff --git a/test/base/utils.py b/test/base/utils.py
index 8df3a2b97..28258e9c3 100644
--- a/test/base/utils.py
+++ b/test/base/utils.py
@@ -1,4 +1,5 @@
 import testbase
+import unittest
 from sqlalchemy import util, sql, exceptions
 from testlib import *
 
@@ -61,6 +62,27 @@ class ColumnCollectionTest(PersistTest):
         cc3.add(c3)
         assert (cc1==cc2).compare(c1 == c2)
         assert not (cc1==cc3).compare(c2 == c3)
+
+class ArgSingletonTest(unittest.TestCase):
+    def test_cleanout(self):
+        util.ArgSingleton.instances.clear()
+        
+        class MyClass(object):
+            __metaclass__ = util.ArgSingleton
+            def __init__(self, x, y):
+                self.x = x
+                self.y = y
+        
+        m1 = MyClass(3, 4)
+        m2 = MyClass(1, 5)
+        m3 = MyClass(3, 4)
+        assert m1 is m3
+        assert m2 is not m3
+        assert len(util.ArgSingleton.instances) == 2
+        
+        m1 = m2 = m3 = None
+        MyClass.dispose(MyClass)
+        assert len(util.ArgSingleton.instances) == 0
         
         
 if __name__ == "__main__":