diff options
author | smiddlek <smiddlek@b1010a0a-674b-0410-b734-77272b80c875> | 2009-03-20 17:03:34 +0000 |
---|---|---|
committer | smiddlek <smiddlek@b1010a0a-674b-0410-b734-77272b80c875> | 2009-03-20 17:03:34 +0000 |
commit | aeaefb32816d68478b63b3e8f84520e4cd7c3a2a (patch) | |
tree | e89b513f88a0890bf8657ba85d7b400717d2c13e /stubout.py | |
parent | 4ea08c31b9acea16a21d751c4ca419346323af11 (diff) | |
download | pymox-aeaefb32816d68478b63b3e8f84520e4cd7c3a2a.tar.gz |
Fix for Issue 5, submitted by agoratim.
Diffstat (limited to 'stubout.py')
-rw-r--r-- | stubout.py | 7 |
1 files changed, 5 insertions, 2 deletions
@@ -118,8 +118,11 @@ class StubOutForTesting: old_child = getattr(parent, child_name) old_attribute = parent.__dict__.get(child_name) - if old_attribute is not None and isinstance(old_attribute, staticmethod): - old_child = staticmethod(old_child) + if old_attribute is not None: + if isinstance(old_attribute, staticmethod): + old_child = staticmethod(old_child) + elif isinstance(old_attribute, classmethod): + old_child = classmethod(old_child.im_func) self.cache.append((parent, old_child, child_name)) setattr(parent, child_name, new_child) |