[edu] 성능 시작측정 코드(decorator, prime, 소수)

[함수 런타임 시간:decorator]

폰트 사이즈 12 고딕

def runtime(func):
    def wrapper(*args, **kwargs):
        import timeit
        start = timeit.default_timer()
        func(*args, **kwargs)
        end = timeit.default_timer()
        print("{}, runtime={}".format(func.__name__, end - start))
    return wrapper


def is_prime(n):
    i = 2
    while i * i <= n:
        if not n % i:
            return False
        i += 1
    return True


def generator_prime(n):
    for i in range(2, n + 1):
        if is_prime(i):
            yield i


@runtime
def count_prime(n):
    count = 0
    for _ in generator_prime(n):
        count += 1
    print(count)