[shapes] shape2opposite uses multisets

Prometheus3375 · Jul 17, 2024 · 570fba7 · 570fba7
1 parent f3b6945
commit 570fba7
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/solve/shapes.py b/solve/shapes.py
@@ -65,19 +65,19 @@ def __sub__(self, other: Shape2D, /) -> Shape2D:
 triangle = Shape2D('triangle')
 square = Shape2D('square')
 
-shape2opposite = {
-    circle:   {triangle, square},
-    triangle: {circle, square},
-    square:   {circle, triangle},
-    }
-
 sphere = Shape3D('sphere', circle, circle)
 pyramid = Shape3D('pyramid', triangle, triangle)
 cube = Shape3D('cube', square, square)
 cone = Shape3D('cone', circle, triangle)
 cylinder = Shape3D('cylinder', circle, square)
 prism = Shape3D('prism', triangle, square)
 
+shape2opposite = {
+    circle:   prism.terms,
+    triangle: cylinder.terms,
+    square:   cone.terms,
+    }
+
 __all__ = (
     'Shape2D',
     'Shape3D',