setup arbitrary test with params

sonos · Jul 12, 2023 · f2b55ee · f2b55ee
1 parent 98385cf
commit f2b55ee
Showing 1 changed file with 9 additions and 2 deletions.
diff --git 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"