Iterate.hs

{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE TemplateHaskell #-}

-- Copyright (C) 2011 John Millikin <jmillikin@gmail.com>
--
-- See license.txt for details
module EnumeratorTests.Text.Iterate
	( test_Iterate
	, test_IterateM
	) where

import           Data.Functor.Identity (runIdentity)
import           Test.Chell

import           Data.Enumerator (($$))
import qualified Data.Enumerator as E
import qualified Data.Enumerator.List as EL
import qualified Data.Enumerator.Text as ET

test_Iterate :: Suite
test_Iterate = assertions "iterate" $ do
	$expect $ equal
		["A", "B", "C"]
		(runIdentity (E.run_ (ET.iterate succ 'A' $$ EL.take 3)))

test_IterateM :: Suite
test_IterateM = assertions "iterateM" $ do
	let succM = return . succ
	$expect $ equal
		["A", "B", "C"]
		(runIdentity (E.run_ (ET.iterateM succM 'A' $$ EL.take 3)))