rainbowadn/rainbowadn/testing/instrument/counter.py
2022-07-10 21:14:02 +03:00

15 lines
368 B
Python

from .instrumentation import *
__all__ = ('Counter',)
class Counter(Instrumentation):
def __init__(self, target, methodname: str):
assert isinstance(methodname, str)
super().__init__(target, methodname)
self.counter = 0
def instrument(self, method, *args, **kwargs):
self.counter += 1
return method(*args, **kwargs)