41 lines
613 B
Python
41 lines
613 B
Python
|
from decorator import Counter
|
||
|
|
||
|
|
||
|
def test_no_args():
|
||
|
@Counter
|
||
|
def say_hello():
|
||
|
print("hello")
|
||
|
|
||
|
say_hello()
|
||
|
say_hello()
|
||
|
say_hello()
|
||
|
say_hello()
|
||
|
|
||
|
assert say_hello.count == 4
|
||
|
|
||
|
|
||
|
def test_arg():
|
||
|
@Counter
|
||
|
def say_hello(name):
|
||
|
print(f"hello {name}")
|
||
|
|
||
|
say_hello("a")
|
||
|
say_hello("b")
|
||
|
say_hello("c")
|
||
|
say_hello("d")
|
||
|
|
||
|
assert say_hello.count == 4
|
||
|
|
||
|
|
||
|
def test_default_arg():
|
||
|
@Counter
|
||
|
def say_hello(name="default"):
|
||
|
print(f"hello {name}")
|
||
|
|
||
|
say_hello("a")
|
||
|
say_hello()
|
||
|
say_hello("c")
|
||
|
say_hello()
|
||
|
|
||
|
assert say_hello.count == 4
|