extend testing

dafny-lang · Mar 20, 2024 · e0c78d4 · e0c78d4
1 parent 07df18f
commit e0c78d4
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 50 deletions.
diff --git a/tests/Tests.dfy b/tests/Tests.dfy
@@ -8,6 +8,7 @@ module Tests {
   import RealArith
   import Helper
   import DafnyVMC
+  import opened Pos
 
   method TestBernoulliIsWithin3SigmaOfTrueMean(
     n: nat,
@@ -119,70 +120,42 @@ module Tests {
     TestEmpiricalIsWithin3SigmaOfTrueMean(n, sum as real, (u - 1) as real / 2.0, (u * u - 1) as real / 12.0, "mean of Uniform(" + Helper.NatToString(u) + ")");
   }
 
-  method TestUniformInterval(n: nat, r: DafnyVMC.Random)
+  method TestUniformInterval(n: nat, a: int, b: int, r: DafnyVMC.Random)
     decreases *
     requires n > 0
+    requires a < b
     modifies r
   {
-    var a := 0;
-    var b := 0;
-    var c := 0;
+    var u := b-a;
+    var arr := new nat[u](i => 0);
     for i := 0 to n {
-      var k := r.UniformIntervalSample(7,10);
-      match k {
-        case 7 => a := a + 1;
-        case 8 => b := b + 1;
-        case 9 => c := c + 1;
-      }
+      var l := r.UniformIntervalSample(a, b);
+      expect a <= l < b, "sample not in the right bound";
+      var k := l-a;
+      assert 0 <= k < u;
+      arr[k] := arr[k] + 1;
     }
-    TestBernoulliIsWithin3SigmaOfTrueMean(n, a as real, 1.0 / 3.0, "p(7)");
-    TestBernoulliIsWithin3SigmaOfTrueMean(n, b as real, 1.0 / 3.0, "p(8)");
-    TestBernoulliIsWithin3SigmaOfTrueMean(n, c as real, 1.0 / 3.0, "p(9)");
-  }
 
-  method TestBernoulli(n: nat, r: DafnyVMC.Random)
-    decreases *
-    requires n > 0
-    modifies r
-  {
-    var t := 0;
-    for i := 0 to n {
-      var b := r.BernoulliSample(1, 5);
-      if b {
-        t := t + 1;
-      }
-    }
-    TestBernoulliIsWithin3SigmaOfTrueMean(n, t as real, 0.2, "p(true)");
-  }
-
-  method TestBernoulli2(n: nat, r: DafnyVMC.Random)
-    decreases *
-    modifies r
-  {
-    var t := 0;
-    for i := 0 to n {
-      var b := r.BernoulliSample(0, 5);
-      if b {
-        t := t + 1;
-      }
+    for i := 0 to u {
+      TestBernoulliIsWithin3SigmaOfTrueMean(n, arr[i] as real, 1.0 / (u as real), "p(" + Helper.NatToString(i) + ")");
     }
 
-    expect t == 0;
   }
 
-  method TestBernoulli3(n: nat, r: DafnyVMC.Random)
+  method TestBernoulli(n: nat, num: nat, den: pos, r: DafnyVMC.Random)
     decreases *
+    requires n > 0
+    requires num <= den
     modifies r
   {
     var t := 0;
     for i := 0 to n {
-      var b := r.BernoulliSample(5, 5);
+      var b := r.BernoulliSample(num, den);
       if b {
         t := t + 1;
       }
     }
-
-    expect t == n;
+    TestBernoulliIsWithin3SigmaOfTrueMean(n, t as real, (num as real) / (den as real), "p(true)");
   }
 
   method TestBernoulliExpNegLe1(n: nat, r: DafnyVMC.Random)

diff --git a/tests/TestsRandom.dfy b/tests/TestsRandom.dfy
@@ -53,33 +53,68 @@ module TestsRandom {
     Tests.TestUniformMean(100_000, NatArith.Power(10, 100), r);
   }
 
+  method {:test} TestUniformInterval0()
+    decreases *
+  {
+    var r := new DafnyVMC.Random();
+    Tests.TestUniformInterval(1_000_000, 0, 1, r);
+  }
 
-  method {:test} TestUniformInterval()
+  method {:test} TestUniformInterval1()
     decreases *
   {
     var r := new DafnyVMC.Random();
-    Tests.TestUniformInterval(1_000_000, r);
+    Tests.TestUniformInterval(1_000_000, -5, 5, r);
   }
 
-  method {:test} TestBernoulli()
+  method {:test} TestUniformInterval2()
     decreases *
   {
     var r := new DafnyVMC.Random();
-    Tests.TestBernoulli(1_000_000, r);
+    Tests.TestUniformInterval(1_000_000, -15, 15, r);
+  }
+
+  method {:test} TestUniformInterval3()
+    decreases *
+  {
+    var r := new DafnyVMC.Random();
+    Tests.TestUniformInterval(1_000_000, -30, -5, r);
+  }
+
+  method {:test} TestUniformInterval4()
+    decreases *
+  {
+    var r := new DafnyVMC.Random();
+    Tests.TestUniformInterval(1_000_000, -30, 30, r);
+  }
+
+  method {:test} TestBernoulli0()
+    decreases *
+  {
+    var r := new DafnyVMC.Random();
+    Tests.TestBernoulli(1_000_000, 1, 5, r);
+  }
+
+  method {:test} TestBernoulli1()
+    decreases *
+  {
+    var r := new DafnyVMC.Random();
+    Tests.TestBernoulli(1_000_000, 0, 5, r);
   }
 
   method {:test} TestBernoulli2()
     decreases *
   {
     var r := new DafnyVMC.Random();
-    Tests.TestBernoulli2(1_000_000, r);
+    Tests.TestBernoulli(1_000_000, 5, 5, r);
   }
 
+
   method {:test} TestBernoulli3()
     decreases *
   {
     var r := new DafnyVMC.Random();
-    Tests.TestBernoulli3(1_000_000, r);
+    Tests.TestBernoulli(1_000_000, 1, 100, r);
   }
 
   method {:test} TestBernoulliExpNegLe1()