]> code.delx.au - offlineimap/blobdiff - testsrc/TestInfrastructure.hs
Made tests use generic types; they don't yet all pass like this
[offlineimap] / testsrc / TestInfrastructure.hs
index 40531b1d1c5283adc90e7ddfd73138ab3ce1264c..45d0f0a18d43733855e04d6f9337c1ad5807a8cf 100644 (file)
@@ -36,13 +36,13 @@ expected @=? actual =
 keysToMap :: Ord k => [k] -> Map.Map k ()
 keysToMap = foldl (\map k -> Map.insert k () map) Map.empty
 
-emptymap :: (Eq k, Ord k) => Map.Map k ()
+emptymap :: (Eq k, Ord k, Show v) => Map.Map k v
 emptymap = Map.empty
 
-instance (Arbitrary k, Eq k, Ord k) => Arbitrary (Map.Map k ()) where
+instance (Arbitrary k, Arbitrary v, Eq k, Ord k) => Arbitrary (Map.Map k v) where
     arbitrary = 
         do items <- arbitrary
-           return $ keysToMap items
+           return $ Map.fromList items
     coarbitrary = coarbitrary . Map.keys
 
 -- Modified from HUnit