blob: 69ef606538c0df77a8a9a6f4ce71bf85b3879281 [file] [log] [blame]
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TemplateHaskell #-}
-- Copyright (C) 2010 John Millikin <jmillikin@gmail.com>
--
-- See license.txt for details
module EnumeratorTests.Text.Zip
( test_Zip
) where
import qualified Control.Exception as Exc
import Data.Functor.Identity (Identity)
import Data.Text (Text)
import Test.Chell
import qualified Data.Enumerator as E
import qualified Data.Enumerator.List as EL
import qualified Data.Enumerator.Text as ET
import EnumeratorTests.Util (equalError)
test_ZipN :: (Eq b, Show b) => Text -> E.Iteratee Text Identity b -> b -> Suite
test_ZipN name iter expected = assertions name $ do
$expect $ equal
expected
(E.runLists_ [[], ["A"], ["B"]] iter)
$([d||])
test_Zip :: Suite
test_Zip = suite "zip"
[ test_ContinueContinue
, test_YieldContinue
, test_ContinueYield
, test_YieldYield
, test_ErrorFirst
, test_ErrorSecond
, test_HandleEOF
, test_Zip3
, test_Zip4
, test_Zip5
, test_Zip6
, test_Zip7
, test_ZipWith
, test_ZipWith3
, test_ZipWith4
, test_ZipWith5
, test_ZipWith6
, test_ZipWith7
]
test_ContinueContinue :: Suite
test_ContinueContinue = assertions "continue-continue" $ do
$expect $ equal
("AB", "AB", ["C"])
(E.runLists_ [["A"], ["B"], ["C"]] $ do
(x, y) <- ET.zip (ET.take 2) (ET.take 2)
extra <- EL.consume
return (x, y, extra))
test_YieldContinue :: Suite
test_YieldContinue = assertions "yield-continue" $ do
$expect $ equal
("A", "AB", ["C"])
(E.runLists_ [["A"], ["B"], ["C"]] $ do
(x, y) <- ET.zip (ET.take 1) (ET.take 2)
extra <- EL.consume
return (x, y, extra))
test_ContinueYield :: Suite
test_ContinueYield = assertions "continue-yield" $ do
$expect $ equal
("AB", "A", ["C"])
(E.runLists_ [["A"], ["B"], ["C"]] $ do
(x, y) <- ET.zip (ET.take 2) (ET.take 1)
extra <- EL.consume
return (x, y, extra))
test_YieldYield :: Suite
test_YieldYield = assertions "yield-yield" $ do
$expect $ equal
("A", "A", ["B", "C"])
(E.runLists_ [["A"], ["B"], ["C"]] $ do
(x, y) <- ET.zip (ET.take 1) (ET.take 1)
extra <- EL.consume
return (x, y, extra))
test_ErrorFirst :: Suite
test_ErrorFirst = assertions "error-first" $ do
$expect $ equalError
(Exc.ErrorCall "error")
(E.runLists [["A"], ["B"], ["C"]] $ do
ET.zip (E.throwError (Exc.ErrorCall "error")) (ET.take 1))
test_ErrorSecond :: Suite
test_ErrorSecond = assertions "error-second" $ do
$expect $ equalError
(Exc.ErrorCall "error")
(E.runLists [["A"], ["B"], ["C"]] $ do
ET.zip (ET.take 1) (E.throwError (Exc.ErrorCall "error")))
test_HandleEOF :: Suite
test_HandleEOF = assertions "handle-eof" $ do
$expect $ equal
("A", "AB", [])
(E.runLists_ [["A"], ["B"]] $ do
(x, y) <- ET.zip (ET.take 1) (ET.take 3)
extra <- EL.consume
return (x, y, extra))
test_Zip3 :: Suite
test_Zip3 = test_ZipN "zip3"
(ET.zip3 ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A')
test_Zip4 :: Suite
test_Zip4 = test_ZipN "zip4"
(ET.zip4 ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A')
test_Zip5 :: Suite
test_Zip5 = test_ZipN "zip5"
(ET.zip5 ET.head ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A', Just 'A')
test_Zip6 :: Suite
test_Zip6 = test_ZipN "zip6"
(ET.zip6 ET.head ET.head ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A', Just 'A', Just 'A')
test_Zip7 :: Suite
test_Zip7 = test_ZipN "zip7"
(ET.zip7 ET.head ET.head ET.head ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A', Just 'A', Just 'A', Just 'A')
test_ZipWith :: Suite
test_ZipWith = test_ZipN "zipWith"
(ET.zipWith (,) ET.head ET.head)
(Just 'A', Just 'A')
test_ZipWith3 :: Suite
test_ZipWith3 = test_ZipN "zipWith3"
(ET.zipWith3 (,,) ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A')
test_ZipWith4 :: Suite
test_ZipWith4 = test_ZipN "zipWith4"
(ET.zipWith4 (,,,) ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A')
test_ZipWith5 :: Suite
test_ZipWith5 = test_ZipN "zipWith5"
(ET.zipWith5 (,,,,) ET.head ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A', Just 'A')
test_ZipWith6 :: Suite
test_ZipWith6 = test_ZipN "zipWith6"
(ET.zipWith6 (,,,,,) ET.head ET.head ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A', Just 'A', Just 'A')
test_ZipWith7 :: Suite
test_ZipWith7 = test_ZipN "zipWith7"
(ET.zipWith7 (,,,,,,) ET.head ET.head ET.head ET.head ET.head ET.head ET.head)
(Just 'A', Just 'A', Just 'A', Just 'A', Just 'A', Just 'A', Just 'A')