1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
|
from __future__ import print_function
import unittest
import wrapt
class Class(object):
def __init__(self, value):
self.value = value
class TestAttributeProxy(unittest.TestCase):
def test_wrap_attribute(self):
wrapt.wrap_object_attribute(__name__, 'Class.value', wrapt.ObjectProxy)
instance = Class(1)
self.assertEqual(instance.value, 1)
self.assertTrue(isinstance(instance.value, wrapt.ObjectProxy))
instance.value = 2
self.assertEqual(instance.value, 2)
self.assertTrue(isinstance(instance.value, wrapt.ObjectProxy))
|