array-builder-0.1.1.0: test/Main.hs

{-# language BangPatterns #-}
{-# language ScopedTypeVariables #-}
{-# language TypeApplications #-}

import Data.Builder (singleton,doubleton,tripleton,run)
import Test.Tasty (defaultMain,testGroup,TestTree)
import Test.Tasty.HUnit ((@=?))
import Data.Semigroup (stimes)
import qualified Data.List as L
import qualified Data.Foldable as F
import qualified Test.Tasty.HUnit as THU

main :: IO ()
main = defaultMain tests

tests :: TestTree
tests = testGroup "Tests"
  [ THU.testCase "A" $ "ABCDEF" @=?
    ( F.toList $ run
      (  singleton 'A'
      <> singleton 'B'
      <> singleton 'C'
      <> singleton 'D'
      <> singleton 'E'
      <> singleton 'F'
      )
    )
  , THU.testCase "B" $ "ABCCCCCCCCCCCCCCCD" @=?
    ( F.toList $ run
      (  singleton 'A'
      <> singleton 'B'
      <> stimes (15 :: Int) (singleton 'C')
      <> singleton 'D'
      )
    )
  , THU.testCase "C" $ (L.replicate 500 'X') @=?
    (F.toList $ run (stimes (500 :: Int) (singleton 'X')))
  , THU.testCase "D" $ "ACDCDCDCDCDCDCDCDCDX" @=?
    ( F.toList $ run
      (  singleton 'A'
      <> stimes (9 :: Int) (doubleton 'C' 'D')
      <> singleton 'X'
      )
    )
  , THU.testCase "E" $ "ABCABCABCABCABCABCABCX" @=?
    ( F.toList $ run
      (  stimes (7 :: Int) (tripleton 'A' 'B' 'C')
      <> singleton 'X'
      )
    )
  ]