Vector2: Support use of weak references

tests/memory: Updated as Vector2 instances are now a tiny bit larger
ppb · Mar 15, 2019 · 525534f · 525534f
1 parent c7a946b
commit 525534f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ppb_vector/vector2.py b/ppb_vector/vector2.py
@@ -57,7 +57,7 @@ class Vector2:
     y: float
 
     # Tell CPython that this isn't an extendable dict
-    __slots__ = ('x', 'y')
+    __slots__ = ('x', 'y', '__weakref__')
 
     @typing.overload
     def __init__(self, x: typing.SupportsFloat, y: typing.SupportsFloat): pass

diff --git a/tests/test_memory.py b/tests/test_memory.py
@@ -20,8 +20,8 @@ def __init__(self, x, y):
 
 @given(x=floats(), y=floats())
 def test_object_size(x, y):
-    """Check that Vector2 is 3 times smaller than a naïve version."""
-    assert sizeof(Vector2(x, y)) < sizeof(DummyVector(x, y)) / 3
+    """Check that Vector2 is 2 times smaller than a naïve version."""
+    assert sizeof(Vector2(x, y)) < sizeof(DummyVector(x, y)) / 2
 
 
 @given(v=vectors())