From be1eb8fbb8cac4ddfd50797d70ddab0b2b306332 Mon Sep 17 00:00:00 2001 From: Graham Dumpleton Date: Tue, 20 Aug 2013 20:59:45 +1000 Subject: [PATCH] Update unit tests to test that isinstance() for base class works via object proxy. --- tests/test_object_proxy.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/tests/test_object_proxy.py b/tests/test_object_proxy.py index 629556b..cd64dec 100644 --- a/tests/test_object_proxy.py +++ b/tests/test_object_proxy.py @@ -8,7 +8,10 @@ import wrapt from wrapt import six OBJECTS_CODE = """ -class Target(object): +class TargetBaseClass(object): + "documentation" + +class Target(TargetBaseClass): "documentation" def target(): @@ -151,6 +154,7 @@ class TestTypeObjectProxy(unittest.TestCase): wrapper = wrapt.ObjectProxy(target) self.assertEqual(wrapper.__class__, target.__class__) + self.assertTrue(isinstance(wrapper, type(target))) def test_class_of_instance(self): @@ -160,7 +164,9 @@ class TestTypeObjectProxy(unittest.TestCase): wrapper = wrapt.ObjectProxy(target) self.assertEqual(wrapper.__class__, target.__class__) - self.assertTrue(isinstance(wrapper, type(target))) + + self.assertTrue(isinstance(wrapper, objects.Target)) + self.assertTrue(isinstance(wrapper, objects.TargetBaseClass)) def test_class_of_function(self): # Test preservation of function __class__ attribute. @@ -169,6 +175,7 @@ class TestTypeObjectProxy(unittest.TestCase): wrapper = wrapt.ObjectProxy(target) self.assertEqual(wrapper.__class__, target.__class__) + self.assertTrue(isinstance(wrapper, type(target))) class TestDirObjectProxy(unittest.TestCase):