Reinstate some tests

haskell · Nov 20, 2023 · 34ecf36 · 34ecf36
1 parent bff27ba
commit 34ecf36
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/test/Spec/Stateful.hs b/test/Spec/Stateful.hs
@@ -180,6 +180,26 @@ statefulSpec =
         ]
     , testGroup "StatefulGen"
         [ statefulSpecFor StateGen unStateGen runStateGenT
+        , statefulSpecFor IOGen unIOGen $ \g action -> do
+            mg <- newIOGenM (unIOGen g)
+            res <- action mg
+            g' <- readGen mg
+            pure (res, g')
+        , statefulSpecFor AtomicGen unAtomicGen $ \g action -> do
+            mg <- newAtomicGenM (unAtomicGen g)
+            res <- action mg
+            g' <- readGen mg
+            pure (res, g')
+        , statefulSpecFor STGen unSTGen $ \g action -> stToIO $ do
+            mg <- newSTGenM (unSTGen g)
+            res <- action mg
+            g' <- readGen mg
+            pure (res, g')
+        , statefulSpecFor TGen unTGen $ \g action -> atomically $ do
+            mg <- newTGenM (unTGen g)
+            res <- action mg
+            g' <- readGen mg
+            pure (res, g')
         ]
     ]