Merge pull request #24 from purescript/more-instances

Add Arb/CoArb instances for Unit and Ordering

README.md

@@ -28,11 +28,11 @@ class Arbitrary t where
   arbitrary :: Gen t
 ```
 
-The `Arbitrary` class represents those types whose values can be 
+The `Arbitrary` class represents those types whose values can be
 _randomly-generated_.
 
 `arbitrary` uses the `Gen` monad to express a random generator for
-the type `t`. Combinators in the `Test.QuickCheck.Gen` 
+the type `t`. Combinators in the `Test.QuickCheck.Gen`
 module can be used to construct random generators.
 
 #### `CoArbitrary`
@@ -138,14 +138,42 @@ instance coarbString :: CoArbitrary String
 ```
 
 
+#### `arbUnit`
+
+``` purescript
+instance arbUnit :: Arbitrary Unit
+```
+
+
+#### `coarbUnit`
+
+``` purescript
+instance coarbUnit :: CoArbitrary Unit
+```
+
+
+#### `arbOrdering`
+
+``` purescript
+instance arbOrdering :: Arbitrary Ordering
+```
+
+
+#### `coarbOrdering`
+
+``` purescript
+instance coarbOrdering :: CoArbitrary Ordering
+```
+
+
 #### `AlphaNumString`
 
 ``` purescript
 newtype AlphaNumString
   = AlphaNumString String
 ```
 
-A newtype for `String` whose `Arbitrary` instance generated random 
+A newtype for `String` whose `Arbitrary` instance generated random
 alphanumeric strings.
 
 #### `arbAlphaNumString`

src/Test/QuickCheck.purs

@@ -35,11 +35,11 @@ import qualified Data.String.Unsafe as SU
 
 import Test.QuickCheck.Gen
 
--- | The `Arbitrary` class represents those types whose values can be 
+-- | The `Arbitrary` class represents those types whose values can be
 -- | _randomly-generated_.
--- | 
+-- |
 -- | `arbitrary` uses the `Gen` monad to express a random generator for
--- | the type `t`. Combinators in the `Test.QuickCheck.Gen` 
+-- | the type `t`. Combinators in the `Test.QuickCheck.Gen`
 -- | module can be used to construct random generators.
 class Arbitrary t where
   arbitrary :: Gen t
@@ -100,7 +100,26 @@ instance arbString :: Arbitrary String where
 instance coarbString :: CoArbitrary String where
   coarbitrary s = coarbitrary $ (S.charCodeAt 0 <$> S.split "" s)
 
--- | A newtype for `String` whose `Arbitrary` instance generated random 
+instance arbUnit :: Arbitrary Unit where
+  arbitrary = return unit
+
+instance coarbUnit :: CoArbitrary Unit where
+  coarbitrary _ = perturbGen 1
+
+instance arbOrdering :: Arbitrary Ordering where
+  arbitrary = do
+    n <- chooseInt 1 3
+    return $ case n of
+      1 -> LT
+      2 -> EQ
+      3 -> GT
+
+instance coarbOrdering :: CoArbitrary Ordering where
+  coarbitrary LT = perturbGen 1
+  coarbitrary EQ = perturbGen 2
+  coarbitrary GT = perturbGen 3
+
+-- | A newtype for `String` whose `Arbitrary` instance generated random
 -- | alphanumeric strings.
 newtype AlphaNumString = AlphaNumString String