diff options
Diffstat (limited to 'test/base/utils.py')
-rw-r--r-- | test/base/utils.py | 22 |
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__": |