abacus/test/AbacusSpec.hs

77 lines
2.1 KiB
Haskell

{-
abacus
Copyright (C) Jonathan Lamothe <jonathan@jlamothe.net>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public
License along with this program. If not, see
<https://www.gnu.org/licenses/>.
-}
module AbacusSpec (spec) where
import Data.Maybe (fromJust)
import Test.Hspec (Spec, context, describe, it, shouldBe)
import Abacus
import Abacus.Internal
spec :: Spec
spec = describe "Abacus" $ do
newAbacusSpec
getNumBeadsSpec
getNumRungsSpec
newAbacusSpec :: Spec
newAbacusSpec = describe "newAbacusSpec" $ mapM_
( \(desc, beads, rungs, expected) -> context desc $ let
actual = newAbacus beads rungs
in it ("should be " ++ show expected) $
actual `shouldBe` expected
)
[ ( "valid", 10, 3, Just $ Abacus 10 [0, 0, 0] )
, ( "no beads", 0, 3, Nothing )
, ( "no rungs", 10, 0, Nothing )
]
getNumBeadsSpec :: Spec
getNumBeadsSpec = describe "getNumBeads" $ mapM_
( \(desc, input, expected) -> context desc $ let
actual = getNumBeads input
in it ("should be " ++ show expected) $
actual `shouldBe` expected
)
[ ( "one bead", oneBead, 1 )
, ( "ten beads", tenBeads, 10 )
] where
oneBead = build 1
tenBeads = build 10
build n = fromJust $ newAbacus n 10
getNumRungsSpec :: Spec
getNumRungsSpec = describe "getNumRungs" $ mapM_
( \(desc, input, expected) -> context desc $ let
actual = getNumRungs input
in it ("should be " ++ show expected) $
actual `shouldBe` expected
)
[ ( "one rung", oneRung, 1 )
, ( "ten rungs", tenRungs, 10 )
] where
oneRung = build 1
tenRungs = build 10
build n = fromJust $ newAbacus 10 n
--jl