Add function Gen.timeSpan

hedgehogqa · Jul 25, 2023 · 19c44e8 · 19c44e8
1 parent f4a4b00
commit 19c44e8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/src/Hedgehog/Gen.fs b/src/Hedgehog/Gen.fs
@@ -513,6 +513,13 @@ module Gen =
             return DateTimeOffset(ticks, TimeSpan.FromMinutes (Operators.float offsetMinutes))
         }
 
+    /// Generates a random TimeSpan using the specified range.
+    let timeSpan (range : Range<TimeSpan>) : Gen<TimeSpan> =
+        range
+        |> Range.map (fun x -> x.Ticks)
+        |> int64
+        |> map TimeSpan.FromTicks
+
     //
     // Sampling
     //

diff --git a/tests/Hedgehog.Tests/GenTests.fs b/tests/Hedgehog.Tests/GenTests.fs
@@ -65,6 +65,20 @@ let genTests = testList "Gen tests" [
         |> List.length
         =! actual.Length
 
+    yield! testCases "timeSpan creates TimeSpan instances"
+        [ 8; 16; 32; 64; 128; 256; 512 ] <| fun count ->
+
+        let actual =
+            (Range.constant TimeSpan.MinValue TimeSpan.MaxValue)
+            |> Gen.timeSpan
+            |> Gen.sample 0 count
+            |> Seq.toList
+
+        actual
+        |> List.distinct
+        |> List.length
+        =! actual.Length
+
     testCase "unicode does not return any surrogate" <| fun _ ->
         let actual =
             Gen.sample 100 100000 Gen.unicode