From 6451d764e050d709227adda2f5cd7956fb8000c1 Mon Sep 17 00:00:00 2001
From: Jan Ferdinand Sauer <ferdinand@neptune.cash>
Date: Wed, 2 Oct 2024 09:52:22 +0200
Subject: [PATCH] feat(Claim): Accept more types for public input

---
 triton-vm/src/proof.rs | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/triton-vm/src/proof.rs b/triton-vm/src/proof.rs
index a2c455bf3..5f1a6aacd 100644
--- a/triton-vm/src/proof.rs
+++ b/triton-vm/src/proof.rs
@@ -69,8 +69,8 @@ impl Claim {
     }
 
     #[must_use]
-    pub fn with_input(mut self, input: Vec<BFieldElement>) -> Self {
-        self.input = input;
+    pub fn with_input(mut self, input: impl Into<Vec<BFieldElement>>) -> Self {
+        self.input = input.into();
         self
     }
 
@@ -90,6 +90,7 @@ mod tests {
     use test_strategy::proptest;
 
     use crate::proof_item::ProofItem;
+    use crate::vm::PublicInput;
 
     use super::*;
 
@@ -100,6 +101,14 @@ mod tests {
         }
     }
 
+    #[test]
+    fn claim_accepts_various_types_for_public_input() {
+        let _claim = Claim::default()
+            .with_input(bfe_vec![42])
+            .with_input(bfe_array![42])
+            .with_input(PublicInput::new(bfe_vec![42]));
+    }
+
     #[proptest]
     fn decode_proof(#[strategy(arb())] proof: Proof) {
         let encoded = proof.encode();