pocketpy/benchmarks/vec.py

import sys

is_cpython = hasattr(sys, 'getrefcount')

if is_cpython:
    class vec2:
        def __init__(self, x, y):
            self.x = x
            self.y = y

        def __add__(self, other):
            return vec2(self.x + other.x, self.y + other.y)

        def __eq__(self, other):
            return self.x == other.x and self.y == other.y
else:
    from linalg import vec2

x = vec2(0, 0)
for i in range(10000000):
    x += vec2(1, 1)

assert x == vec2(10000000, 10000000)