From f2b55eefb40a7dcec44fb2def7813387a0c34485 Mon Sep 17 00:00:00 2001
From: Mathieu Poumeyrol <kali@zoy.org>
Date: Mon, 10 Jul 2023 10:42:22 +0200
Subject: [PATCH] setup arbitrary test with params

---
 test-rt/test-tflite/suite.rs | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/test-rt/test-tflite/suite.rs b/test-rt/test-tflite/suite.rs
index f30f2a536b..61d6ed6403 100644
--- a/test-rt/test-tflite/suite.rs
+++ b/test-rt/test-tflite/suite.rs
@@ -1,8 +1,14 @@
+use suite_conv::conv_f32::{ConvProblem, ConvProblemParams};
+
 pub fn suite() -> infra::TestSuite {
     let mut onnx = suite_onnx::suite().clone();
     onnx.ignore(&ignore_onnx);
     let mut conv = suite_conv::suite().unwrap().clone();
     conv.ignore(&ignore_conv);
+    conv.add_arbitrary::<ConvProblem>(
+        "proptest",
+        ConvProblemParams { no_arbitrary_grouping: true, ..ConvProblemParams::default() },
+    );
     infra::TestSuite::default().with("onnx", onnx).with("conv", conv)
 }
 
@@ -13,11 +19,12 @@ fn ignore_onnx(t: &[String]) -> bool {
 
 fn ignore_conv(t: &[String]) -> bool {
     let unit: &str = t.last().map(|s| &**s).unwrap();
-    t[0] == "q" 
+    t[0] == "q"
+        || unit == "proptest"
         // grouping and depthwise
         || unit == "depthwise_0"
         || unit.starts_with("group")
-        // conv 3D 
+        // conv 3D
         || unit == "lazy_im2col_big"
         || unit == "lazy_im2col_big_2"
         || unit == "batch_3d"