new_type_demo.py

from typing import NewType
import timeit


UserID = NewType('UserID', int)


def check_on(user_id: UserID) -> int:
    return user_id * 2


def check_on_all_users(n: int):
    acc = 0
    for i in range(n):
        acc += check_on(UserID(i))
    return acc


def check_on_all_users_without_cast(n: int):
    acc = 0
    for i in range(n):
        acc += check_on(i)
    return acc


if __name__ == '__main__':
    print("Running timeit")
    print(timeit.timeit("check_on_all_users(10_000)", globals=locals(), number=1_000))
    print(timeit.timeit("check_on_all_users_without_cast(10_000)", globals=locals(), number=1_000))