zio · mijicd · Nov 24, 2020 · Nov 24, 2020 · Nov 24, 2020 · Nov 24, 2020
diff --git a/core-tests/shared/src/test/scala/zio/stm/random/TRandomSpec.scala b/core-tests/shared/src/test/scala/zio/stm/random/TRandomSpec.scala
@@ -0,0 +1,74 @@
+package zio.stm.random
+
+import zio.ZIOBaseSpec
+import zio.random.Random
+import zio.test.Assertion._
+import zio.test._
+
+object TRandomSpec extends ZIOBaseSpec {
+
+  implicit val DoubleOrdering: Ordering[Double] =
+    (l, r) => java.lang.Double.compare(l, r)
+
+  implicit val FloatOrdering: Ordering[Float] =
+    (l, r) => java.lang.Float.compare(l, r)
+
+  def spec: ZSpec[Environment, Failure] = suite("TRandomSpec")(
+    testM("nextDoubleBetween generates doubles in specified range") {
+      checkM(genDoubles) { case (min, max) =>
+        for {
+          n <- nextDoubleBetween(min, max).commit
+        } yield assert(n)(isGreaterThanEqualTo(min)) &&
+          assert(n)(isLessThan(max))
+      }
+    },
+    testM("nextFloatBetween generates floats in specified range") {
+      checkM(genFloats) { case (min, max) =>
+        for {
+          n <- nextFloatBetween(min, max).commit
+        } yield assert(n)(isGreaterThanEqualTo(min)) &&
+          assert(n)(isLessThan(max))
+      }
+    },
+    testM("nextIntBetween generates integers in specified range") {
+      checkM(genInts) { case (min, max) =>
+        for {
+          n <- nextIntBetween(min, max).commit
+        } yield assert(n)(isGreaterThanEqualTo(min)) &&
+          assert(n)(isLessThan(max))
+      }
+    },
+    testM("nextLongBetween generates longs in specified range") {
+      checkM(genLongs) { case (min, max) =>
+        for {
+          n <- nextLongBetween(min, max).commit
+        } yield assert(n)(isGreaterThanEqualTo(min)) &&
+          assert(n)(isLessThan(max))
+      }
+    }
+  ).provideCustomLayer(TRandom.live)
+
+  val genDoubles: Gen[Random, (Double, Double)] =
+    for {
+      a <- Gen.anyDouble
+      b <- Gen.anyDouble if a != b
+    } yield if (b > a) (a, b) else (b, a)
+
+  val genFloats: Gen[Random, (Float, Float)] =
+    for {
+      a <- Gen.anyFloat
+      b <- Gen.anyFloat if a != b
+    } yield if (b > a) (a, b) else (b, a)
+
+  val genInts: Gen[Random, (Int, Int)] =
+    for {
+      a <- Gen.anyInt
+      b <- Gen.anyInt if a != b
+    } yield if (b > a) (a, b) else (b, a)
+
+  val genLongs: Gen[Random, (Long, Long)] =
+    for {
+      a <- Gen.anyLong
+      b <- Gen.anyLong if a != b
+    } yield if (b > a) (a, b) else (b, a)
+}
diff --git a/core/shared/src/main/scala/zio/random/package.scala b/core/shared/src/main/scala/zio/random/package.scala
@@ -165,7 +165,7 @@ package object random {
   }
 
   /**
-   * Generates a pseudo-random boolean.
+   * generates a pseudo-random boolean.
    */
   val nextBoolean: URIO[Random, Boolean] =
     ZIO.accessM(_.get.nextBoolean)

diff --git a/core/shared/src/main/scala/zio/stm/package.scala b/core/shared/src/main/scala/zio/stm/package.scala
@@ -25,4 +25,5 @@ package object stm extends EitherCompat {
   type TaskSTM[+A]   = ZSTM[Any, Throwable, A]
   type TRef[A]       = ZTRef[Nothing, Nothing, A, A]
   type ETRef[+E, A]  = ZTRef[E, E, A, A]
+
 }