init

tests/test_plane.py

@@ -0,0 +1,37 @@
+import numpy as np
+
+from plane import Plane
+from ray import Ray
+from vector import Vector
+
+
+def test_plane():
+  p = Plane(
+    Vector.fromargs(0, 0, 0),
+    Vector.fromargs(0, 0, 1),
+  )
+  assert (p & Vector.fromargs(0, 0, 0))
+  assert (p & Vector.fromargs(1, 1, 0))
+  assert not (p & Vector.fromargs(0, 0, -1))
+
+  p = Plane(
+    Vector.fromargs(1, 1),
+    Vector.fromargs(1, 1),
+  )
+
+  assert not (p & Vector.fromargs(0, 0))
+  assert (p & Vector.fromargs(2, 0))
+
+
+def test_reflection():
+  p = Plane(
+    Vector.fromargs(0, 0),
+    Vector.fromargs(0, 1),
+  )
+
+  r = Ray(
+    Vector.fromargs(-1, 1),
+    Vector.fromargs(1, -1),
+  )
+
+  assert p.intersect(r) == np.sqrt(2)