Update src/algorithm/iter.rs

Co-authored-by: Seyon Sivarajah <[email protected]>
CQCL · Feb 12, 2024 · 768d07a · 768d07a
1 parent e454dab
commit 768d07a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/src/algorithm/iter.rs b/src/algorithm/iter.rs
@@ -209,14 +209,11 @@ mod test {
     #[test]
     fn out_of_order() -> Result<(), Box<dyn Error>> {
         use crate::ops::handle::NodeHandle;
-        let sig = crate::types::PolyFuncType::new(
-            [],
-            FunctionType::new(type_row![USIZE_T], type_row![USIZE_T]),
-        );
-        let mut fun_builder = FunctionBuilder::new("f3", sig.clone())?;
+        let sig = FunctionType::new_endo(type_row![USIZE_T]);
+        let mut fun_builder = FunctionBuilder::new("f3", sig.clone().into())?;
         let [i] = fun_builder.input_wires_arr();
         let noop = fun_builder.add_dataflow_op(ops::LeafOp::Noop { ty: USIZE_T }, [i])?;
-        let dfg_builder = fun_builder.dfg_builder(sig.body().clone(), None, [noop.out_wire(0)])?;
+        let dfg_builder = fun_builder.dfg_builder(sig.clone(), None, [noop.out_wire(0)])?;
         let [i1] = dfg_builder.input_wires_arr();
         let dfg = dfg_builder.finish_with_outputs([i1])?;
         let mut h = fun_builder.finish_prelude_hugr_with_outputs([dfg.out_wire(0)])?;