Add {left,right}_moves in python bindings

t4ccer · Nov 5, 2023 · 6446b79 · 6446b79
1 parent be10498
commit 6446b79
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/cgt-py/src/domineering.rs b/cgt-py/src/domineering.rs
@@ -62,4 +62,20 @@ impl PyDomineering {
             ),
         }
     }
+
+    fn left_moves(&self) -> Vec<Self> {
+        self.inner
+            .left_moves()
+            .into_iter()
+            .map(Self::from)
+            .collect()
+    }
+
+    fn right_moves(&self) -> Vec<Self> {
+        self.inner
+            .right_moves()
+            .into_iter()
+            .map(Self::from)
+            .collect()
+    }
 }
diff --git a/cgt-py/src/ski_jumps.rs b/cgt-py/src/ski_jumps.rs
@@ -51,4 +51,20 @@ impl PySkiJumps {
             ),
         }
     }
+
+    fn left_moves(&self) -> Vec<Self> {
+        self.inner
+            .left_moves()
+            .into_iter()
+            .map(Self::from)
+            .collect()
+    }
+
+    fn right_moves(&self) -> Vec<Self> {
+        self.inner
+            .right_moves()
+            .into_iter()
+            .map(Self::from)
+            .collect()
+    }
 }