array-builder-0.1.2.0: test/Main.hs

{-# language BangPatterns #-}
{-# language PatternSynonyms #-}
{-# 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 Data.Builder.Catenable (pattern (:<), pattern (:>))

import qualified Data.Builder.Catenable as Cat
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"
  [ testGroup "Data.Builder"
    [ 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'
        )
      )
    ]
  , testGroup "Data.Builder.Catenable"
    [ THU.testCase "A" $ "ABCDEF" @=?
      ( F.toList $ Cat.run
        ( ('A' :< 'B' :< 'C' :< mempty)
          <>
          (mempty :> 'D' :> 'E' :> 'F')
        )
      )
    , THU.testCase "B" $ "DEF" @=?
      (F.toList $ Cat.run (mempty :> 'D' :> 'E' :> 'F'))
    ]
  ]