timeit_proto_compare_with_is.py

import random

from multiaddr.protocols import (
    Protocol,
    PROTOCOLS,
    P_IP6ZONE,
)

protocols = tuple(filter(lambda x: x.code != P_IP6ZONE, PROTOCOLS))


def compare_with_is(a: Protocol, b: Protocol) -> bool:
    return a is b or a == b


def compare(a: Protocol, b: Protocol) -> bool:
    return a == b


def test(func_compare):
    code = random.choice(protocols)
    for _code in protocols:
        if func_compare(code, _code):
            return


if __name__ == "__main__":
    import timeit
    print(timeit.timeit("test(compare_with_is)", setup="from __main__ import test, compare_with_is"))
    print(timeit.timeit("test(compare)", setup="from __main__ import test, compare"))