diff --git a/NightLight.Core.Tests/GenHelpers.fs b/NightLight.Core.Tests/GenHelpers.fs index 9d8bbf2..7570b7e 100644 --- a/NightLight.Core.Tests/GenHelpers.fs +++ b/NightLight.Core.Tests/GenHelpers.fs @@ -7,3 +7,5 @@ let concatGens (gens: Gen<'a list> list) : Gen<'a list> = match gens with | [] -> Gen.constant [] | first :: rest -> rest |> List.fold (fun accGen g -> Gen.map2 (@) accGen g) first + +let (=<<) m f = Gen.bind m f diff --git a/NightLight.Core.Tests/NightLightTests.fs b/NightLight.Core.Tests/NightLightTests.fs index d622aa3..8a96736 100644 --- a/NightLight.Core.Tests/NightLightTests.fs +++ b/NightLight.Core.Tests/NightLightTests.fs @@ -40,7 +40,7 @@ type NightLightTests() = [] let ``All lights that are on should be white or yellow during the day`` () = concatGens - [ Gen.bind genInitialInteractionsAndEndWith genTimeChangedToDay + [ genInitialInteractionsAndEndWith =<< genTimeChangedToDay genInteractionsExcept isTimeChangedToNight ] |> Arb.fromGen |> Prop.forAll @@ -53,7 +53,7 @@ type NightLightTests() = [] let ``All lights that are on should be red during the night`` () = concatGens - [ Gen.bind genInitialInteractionsAndEndWith genTimeChangedToNight + [ genInitialInteractionsAndEndWith =<< genTimeChangedToNight genInteractionsExcept isTimeChangedToDay ] |> Arb.fromGen |> Prop.forAll @@ -105,9 +105,9 @@ type NightLightTests() = = let genInitialInteractionsListThatEndsWithTransitionToDay = concatGens - [ Gen.bind genInitialInteractionsAndEndWith genTimeChangedToNight + [ genInitialInteractionsAndEndWith =<< genTimeChangedToNight genInteractionsExcept isTimeChangedToDay - Gen.map List.singleton genTimeChangedToDay ] + genTimeChangedToDay |> Gen.map List.singleton ] concatGens [ genInitialInteractionsListThatEndsWithTransitionToDay