Add --max-success for running more or less prop runs on demand.

crates/aiken-project/src/lib.rs

@@ -223,6 +223,7 @@ where
         Ok(())
     }
 
+    #[allow(clippy::too_many_arguments)]
     pub fn check(
         &mut self,
         skip_tests: bool,
@@ -230,6 +231,7 @@ where
         verbose: bool,
         exact_match: bool,
         seed: u32,
+        property_max_success: usize,
         tracing: Tracing,
     ) -> Result<(), Vec<Error>> {
         let options = Options {
@@ -242,6 +244,7 @@ where
                     verbose,
                     exact_match,
                     seed,
+                    property_max_success,
                 }
             },
         };
@@ -328,6 +331,7 @@ where
                 verbose,
                 exact_match,
                 seed,
+                property_max_success,
             } => {
                 let tests =
                     self.collect_tests(verbose, match_tests, exact_match, options.tracing)?;
@@ -336,7 +340,7 @@ where
                     self.event_listener.handle_event(Event::RunningTests);
                 }
 
-                let tests = self.run_tests(tests, seed);
+                let tests = self.run_tests(tests, seed, property_max_success);
 
                 self.checks_count = if tests.is_empty() {
                     None
@@ -848,7 +852,12 @@ where
         Ok(tests)
     }
 
-    fn run_tests(&self, tests: Vec<Test>, seed: u32) -> Vec<TestResult<UntypedExpr, UntypedExpr>> {
+    fn run_tests(
+        &self,
+        tests: Vec<Test>,
+        seed: u32,
+        property_max_success: usize,
+    ) -> Vec<TestResult<UntypedExpr, UntypedExpr>> {
         use rayon::prelude::*;
 
         let data_types = utils::indexmap::as_ref_values(&self.data_types);
@@ -857,7 +866,7 @@ where
             .into_par_iter()
             .map(|test| match test {
                 Test::UnitTest(unit_test) => unit_test.run(),
-                Test::PropertyTest(property_test) => property_test.run(seed),
+                Test::PropertyTest(property_test) => property_test.run(seed, property_max_success),
             })
             .collect::<Vec<TestResult<(Constant, Rc<Type>), PlutusData>>>()
             .into_iter()

crates/aiken-project/src/options.rs

@@ -11,6 +11,7 @@ pub enum CodeGenMode {
         verbose: bool,
         exact_match: bool,
         seed: u32,
+        property_max_success: usize,
     },
     Build(bool),
     NoOp,

crates/aiken-project/src/test_framework.rs

@@ -214,13 +214,11 @@ pub struct Fuzzer<T> {
 }
 
 impl PropertyTest {
-    const MAX_TEST_RUN: usize = 100;
+    pub const DEFAULT_MAX_SUCCESS: usize = 100;
 
-    /// Run a property test from a given seed. The property is run at most MAX_TEST_RUN times. It
+    /// Run a property test from a given seed. The property is run at most DEFAULT_MAX_SUCCESS times. It
     /// may stops earlier on failure; in which case a 'counterexample' is returned.
-    pub fn run<U>(self, seed: u32) -> TestResult<U, PlutusData> {
-        let n = PropertyTest::MAX_TEST_RUN;
-
+    pub fn run<U>(self, seed: u32, n: usize) -> TestResult<U, PlutusData> {
         let mut labels = BTreeMap::new();
 
         let (counterexample, iterations) =
@@ -1378,7 +1376,7 @@ mod test {
         fn expect_failure(&self) -> Counterexample {
             let mut labels = BTreeMap::new();
             match self.run_n_times(
-                PropertyTest::MAX_TEST_RUN,
+                PropertyTest::DEFAULT_MAX_SUCCESS,
                 Prng::from_seed(42),
                 None,
                 &mut labels,
@@ -1397,7 +1395,9 @@ mod test {
             }
         "#});
 
-        assert!(prop.run::<()>(42).is_success());
+        assert!(prop
+            .run::<()>(42, PropertyTest::DEFAULT_MAX_SUCCESS)
+            .is_success());
     }
 
     #[test]
@@ -1419,7 +1419,7 @@ mod test {
             }
         "#});
 
-        match prop.run::<()>(42) {
+        match prop.run::<()>(42, PropertyTest::DEFAULT_MAX_SUCCESS) {
             TestResult::UnitTestResult(..) => unreachable!("property returned unit-test result ?!"),
             TestResult::PropertyTestResult(result) => {
                 assert!(