diff --git a/benchmarks/vec.py b/benchmarks/vec.py new file mode 100644 index 00000000..fbb959d4 --- /dev/null +++ b/benchmarks/vec.py @@ -0,0 +1,23 @@ +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(5000000, 5000000)