Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from unittest.mock import patch
- def spy(target, attribute, **kwargs):
- if kwargs.get('wraps') is None:
- kwargs['wraps'] = getattr(target, attribute)
- return patch.object(target, attribute , wraps = kwargs['wraps'])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement