tests for HashSetUtils

Author: danslapman

test/HashSetUtilsSpec.hs

@@ -0,0 +1,23 @@
+module HashSetUtilsSpec (hashSetTestCases) where
+
+import qualified Data.HashMap.Strict as HM
+import Data.HashSet
+import HashSetUtils
+import Test.HUnit
+
+testMaybeNesEmptyHashSet :: Test
+testMaybeNesEmptyHashSet = TestCase $ assertEqual "maybeNes empty HashSet" Nothing (maybeNes (empty :: HashSet Int))
+
+testMaybeNesNonEmptyHashSet :: Test
+testMaybeNesNonEmptyHashSet = TestCase $ assertEqual "maybeNes non-empty HashSet" (Just $ (fromList [1, 2, 3] :: HashSet Int)) (maybeNes $ fromList [1, 2, 3])
+
+fromHashMapTest :: Test
+fromHashMapTest = TestCase $ assertEqual "fromHashMap" (fromList [(1, "a"), (2, "b"), (3, "c")] :: HashSet (Int, String)) (fromHashMap $ HM.fromList [(1, "a"), (2, "b"), (3, "c")])
+
+hashSetTestCases :: Test
+hashSetTestCases =
+  TestList
+    [ testMaybeNesEmptyHashSet,
+      testMaybeNesNonEmptyHashSet,
+      fromHashMapTest
+    ]

test/Spec.hs

@@ -1,10 +1,17 @@
+import HashSetUtilsSpec
 import SequenceUtilsSpec
 import Test.HUnit
-import System.Exit (exitFailure, exitSuccess)
+import System.Exit (exitFailure)
+
+processCounts :: Counts -> IO ()
+processCounts counts = do
+  if errors counts + failures counts == 0
+  then return ()
+  else exitFailure
 
 main :: IO ()
 main = do
-  counts <- runTestTT seqTestCases
-  if errors counts + failures counts == 0
-        then exitSuccess
-        else exitFailure
+  hsCounts <- runTestTT hashSetTestCases
+  processCounts hsCounts
+  seqCounts <- runTestTT seqTestCases
+  processCounts seqCounts