test_reflect_prop: Add a property tying reflect() and angle()

ppb · Oct 13, 2018 · 6e18908 · 6e18908
1 parent f5cc638
commit 6e18908
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_vector2_reflect.py b/tests/test_vector2_reflect.py
@@ -2,7 +2,7 @@
 import pytest
 from hypothesis import given, assume, note
 from math import isclose, isinf
-from utils import units, vectors
+from utils import angle_isclose, units, vectors
 
 
 reflect_data = (
@@ -28,3 +28,6 @@ def test_reflect_prop(initial: Vector2, normal: Vector2):
     assert not any(map(isinf, reflected))
     assert initial.isclose(returned)
     assert isclose((initial * normal), -(reflected * normal))
+    assert angle_isclose(normal.angle(initial),
+                         180 - normal.angle(reflected)
+    )