@@ -9,9 +9,9 @@ module Tests.Properties
99 ) where
1010
1111import Control.Applicative ((<$>), (<*>))
12−import Control.Arrow ((***), second)
12+import Control.Arrow ((***), first, second)
1313import Data.Bits ((.&.))
14−import Data.Char (chr, isDigit, isHexDigit, isLower, isSpace, isUpper, ord)
14+import Data.Char (chr, isDigit, isHexDigit, isLower, isSpace, isLetter, isUpper, ord)
1515import Data.Int (Int8, Int16, Int32, Int64)
1616import Data.Monoid (Monoid(..))
1717import Data.String (IsString(fromString))
@@ -23,6 +23,7 @@ import Data.Text.Internal.Search (indices)
2323import Data.Text.Lazy.Read as TL
2424import Data.Text.Read as T
2525import Data.Word (Word, Word8, Word16, Word32, Word64)
26+import Data.Maybe (mapMaybe)
2627import Numeric (showEFloat, showFFloat, showGFloat, showHex)
2728import Prelude hiding (replicate)
2829import Test.Framework (Test, testGroup)
@@ -30,6 +31,7 @@ import Test.Framework.Providers.QuickCheck2 (testProperty)
3031import Test.QuickCheck hiding ((.&.))
3132import Test.QuickCheck.Monadic
3233import Test.QuickCheck.Property (Property(..))
34+import Test.QuickCheck.Unicode (char)
3335import Tests.QuickCheckUtils
3436import Tests.Utils
3537import Text.Show.Functions ()
@@ -37,6 +39,7 @@ import qualified Control.Exception as Exception
3739import qualified Data.Bits as Bits (shiftL, shiftR)
3840import qualified Data.ByteString as B
3941import qualified Data.ByteString.Lazy as BL
42+import qualified Data.Char as C
4043import qualified Data.List as L
4144import qualified Data.Text as T
4245import qualified Data.Text.Encoding as E
@@ -60,10 +63,10 @@ t_pack_unpack = (T.unpack . T.pack) `eq` id
6063tl_pack_unpack = (TL.unpack . TL.pack) `eq` id
6164t_stream_unstream = (S.unstream . S.stream) `eq` id
6265tl_stream_unstream = (SL.unstream . SL.stream) `eq` id
63−t_reverse_stream t = (S.reverse . S.reverseStream) t == t
64−t_singleton c = [c] == (T.unpack . T.singleton) c
65−tl_singleton c = [c] == (TL.unpack . TL.singleton) c
66−tl_unstreamChunks x = f 11 x == f 1000 x
66+t_reverse_stream t = (S.reverse . S.reverseStream) t === t
67+t_singleton c = [c] === (T.unpack . T.singleton) c
68+tl_singleton c = [c] === (TL.unpack . TL.singleton) c
69+tl_unstreamChunks x = f 11 x === f 1000 x
6770 where f n = SL.unstreamChunks n . S.streamList
6871tl_chunk_unchunk = (TL.fromChunks . TL.toChunks) `eq` id
6972tl_from_to_strict = (TL.fromStrict . TL.toStrict) `eq` id
@@ -74,13 +77,13 @@ encodeL1 = B.pack . map (fromIntegral . fromEnum) . T.unpack
7477encodeLazyL1 :: TL.Text -> BL.ByteString
7578encodeLazyL1 = BL.fromChunks . map encodeL1 . TL.toChunks
7679
77−t_ascii t = E.decodeASCII (E.encodeUtf8 a) == a
80+t_ascii t = E.decodeASCII (E.encodeUtf8 a) === a
7881 where a = T.map (\c -> chr (ord c `mod` 128)) t
79−tl_ascii t = EL.decodeASCII (EL.encodeUtf8 a) == a
82+tl_ascii t = EL.decodeASCII (EL.encodeUtf8 a) === a
8083 where a = TL.map (\c -> chr (ord c `mod` 128)) t
81−t_latin1 t = E.decodeLatin1 (encodeL1 a) == a
84+t_latin1 t = E.decodeLatin1 (encodeL1 a) === a
8285 where a = T.map (\c -> chr (ord c `mod` 256)) t
83−tl_latin1 t = EL.decodeLatin1 (encodeLazyL1 a) == a
86+tl_latin1 t = EL.decodeLatin1 (encodeLazyL1 a) === a
8487 where a = TL.map (\c -> chr (ord c `mod` 256)) t
8588t_utf8 = forAll genUnicode $ (E.decodeUtf8 . E.encodeUtf8) `eq` id
8689t_utf8' = forAll genUnicode $ (E.decodeUtf8' . E.encodeUtf8) `eq` (id . Right)
@@ -111,7 +114,7 @@ t_utf8_undecoded = forAll genUnicode $ \t ->
111114 let b = E.encodeUtf8 t
112115 ls = concatMap (leftover . E.encodeUtf8 . T.singleton) . T.unpack $ t
113116 leftover = (++ [B.empty]) . init . tail . B.inits
114− in (map snd . feedChunksOf 1 E.streamDecodeUtf8) b == ls
117+ in (map snd . feedChunksOf 1 E.streamDecodeUtf8) b === ls
115118
116119data Badness = Solo | Leading | Trailing
117120 deriving (Eq, Show)
@@ -201,8 +204,8 @@ t_mconcat = unsquare $
201204 mconcat `eq` (unpackS . mconcat . L.map T.pack)
202205tl_mconcat = unsquare $
203206 mconcat `eq` (unpackS . mconcat . L.map TL.pack)
204−t_mempty = mempty == (unpackS (mempty :: T.Text))
205−tl_mempty = mempty == (unpackS (mempty :: TL.Text))
207+t_mempty = mempty === (unpackS (mempty :: T.Text))
208+tl_mempty = mempty === (unpackS (mempty :: TL.Text))
206209t_IsString = fromString `eqP` (T.unpack . fromString)
207210tl_IsString = fromString `eqP` (TL.unpack . fromString)
208211
@@ -229,6 +232,13 @@ sf_uncons p = (uncons . L.filter p) `eqP`
229232 (fmap (second unpackS) . S.uncons . S.filter p)
230233t_uncons = uncons `eqP` (fmap (second unpackS) . T.uncons)
231234tl_uncons = uncons `eqP` (fmap (second unpackS) . TL.uncons)
235+
236+unsnoc xs@(_:_) = Just (init xs, last xs)
237+unsnoc [] = Nothing
238+
239+t_unsnoc = unsnoc `eqP` (fmap (first unpackS) . T.unsnoc)
240+tl_unsnoc = unsnoc `eqP` (fmap (first unpackS) . TL.unsnoc)
241+
232242s_head = head `eqP` S.head
233243sf_head p = (head . L.filter p) `eqP` (S.head . S.filter p)
234244t_head = head `eqP` T.head
@@ -326,6 +336,19 @@ t_toUpper_upper t = p (T.toUpper t) >= p t
326336 where p = T.length . T.filter isUpper
327337tl_toUpper_upper t = p (TL.toUpper t) >= p t
328338 where p = TL.length . TL.filter isUpper
339+t_toTitle_title t = all (<= 1) (caps w)
340+ where caps = fmap (T.length . T.filter isUpper) . T.words . T.toTitle
341+ -- TIL: there exist uppercase-only letters
342+ w = T.filter (\c -> if C.isUpper c then C.toLower c /= c else True) t
343+t_toTitle_1stNotLower = and . notLow . T.toTitle . T.filter stable
344+ where notLow = mapMaybe (fmap (not . isLower) . (T.find isLetter)) . T.words
345+ -- Surprise! The Spanish/Portuguese ordinal indicators changed
346+ -- from category Ll (letter, lowercase) to Lo (letter, other)
347+ -- in Unicode 7.0
348+ -- Oh, and there exist lowercase-only letters (see previous test)
349+ stable c = if isLower c
350+ then C.toUpper c /= c
351+ else c /= '\170' && c /= '\186'
329352
330353justifyLeft k c xs = xs ++ L.replicate (k - length xs) c
331354justifyRight m n xs = L.replicate (m - length xs) n ++ xs
@@ -514,12 +537,20 @@ s_takeWhile p = L.takeWhile p `eqP` (unpackS . S.takeWhile p)
514537s_takeWhile_s p = L.takeWhile p `eqP` (unpackS . S.unstream . S.takeWhile p)
515538sf_takeWhile q p = (L.takeWhile p . L.filter q) `eqP`
516539 (unpackS . S.takeWhile p . S.filter q)
540+noMatch = do
541+ c <- char
542+ d <- suchThat char (/= c)
543+ return (c,d)
517544t_takeWhile p = L.takeWhile p `eqP` (unpackS . T.takeWhile p)
518545tl_takeWhile p = L.takeWhile p `eqP` (unpackS . TL.takeWhile p)
519546t_takeWhileEnd p = (L.reverse . L.takeWhile p . L.reverse) `eqP`
520547 (unpackS . T.takeWhileEnd p)
548+t_takeWhileEnd_null t = forAll noMatch $ \(c,d) -> T.null $
549+ T.takeWhileEnd (==d) (T.snoc t c)
521550tl_takeWhileEnd p = (L.reverse . L.takeWhile p . L.reverse) `eqP`
522551 (unpackS . TL.takeWhileEnd p)
552+tl_takeWhileEnd_null t = forAll noMatch $ \(c,d) -> TL.null $
553+ TL.takeWhileEnd (==d) (TL.snoc t c)
523554s_dropWhile p = L.dropWhile p `eqP` (unpackS . S.dropWhile p)
524555s_dropWhile_s p = L.dropWhile p `eqP` (unpackS . S.unstream . S.dropWhile p)
525556sf_dropWhile q p = (L.dropWhile p . L.filter q) `eqP`
@@ -720,7 +751,7 @@ tl_indices (NotEmpty s) = lazyIndices s `eq` S.indices s
720751 conc = T.concat . TL.toChunks
721752t_indices_occurs = unsquare $ \(NotEmpty t) ts ->
722753 let s = T.intercalate t ts
723− in Slow.indices t s == indices t s
754+ in Slow.indices t s === indices t s
724755
725756-- Bit shifts.
726757shiftL w = forAll (choose (0,width-1)) $ \k -> Bits.shiftL w k == U.shiftL w k
@@ -812,18 +843,18 @@ tb_formatRealFloat_double (a::Double) = tb_formatRealFloat a
812843-- Reading.
813844
814845t_decimal (n::Int) s =
815− T.signed T.decimal (T.pack (show n) `T.append` t) == Right (n,t)
846+ T.signed T.decimal (T.pack (show n) `T.append` t) === Right (n,t)
816847 where t = T.dropWhile isDigit s
817848tl_decimal (n::Int) s =
818− TL.signed TL.decimal (TL.pack (show n) `TL.append` t) == Right (n,t)
849+ TL.signed TL.decimal (TL.pack (show n) `TL.append` t) === Right (n,t)
819850 where t = TL.dropWhile isDigit s
820851t_hexadecimal m s ox =
821− T.hexadecimal (T.concat [p, T.pack (showHex n ""), t]) == Right (n,t)
852+ T.hexadecimal (T.concat [p, T.pack (showHex n ""), t]) === Right (n,t)
822853 where t = T.dropWhile isHexDigit s
823854 p = if ox then "0x" else ""
824855 n = getPositive m :: Int
825856tl_hexadecimal m s ox =
826− TL.hexadecimal (TL.concat [p, TL.pack (showHex n ""), t]) == Right (n,t)
857+ TL.hexadecimal (TL.concat [p, TL.pack (showHex n ""), t]) === Right (n,t)
827858 where t = TL.dropWhile isHexDigit s
828859 p = if ox then "0x" else ""
829860 n = getPositive m :: Int
@@ -867,11 +898,11 @@ tl_write_read_line e m b t = write_read head TL.filter TL.hPutStrLn
867898
868899t_dropWord16 m t = dropWord16 m t `T.isSuffixOf` t
869900t_takeWord16 m t = takeWord16 m t `T.isPrefixOf` t
870−t_take_drop_16 m t = T.append (takeWord16 n t) (dropWord16 n t) == t
901+t_take_drop_16 m t = T.append (takeWord16 n t) (dropWord16 n t) === t
871902 where n = small m
872903t_use_from t = monadicIO $ assert . (==t) =<< run (useAsPtr t fromPtr)
873904
874−t_copy t = T.copy t == t
905+t_copy t = T.copy t === t
875906
876907-- Regression tests.
877908s_filter_eq s = S.filter p t == S.streamList (filter p s)
@@ -967,6 +998,8 @@ tests =
967998 testProperty "sf_uncons" sf_uncons,
968999 testProperty "t_uncons" t_uncons,
9691000 testProperty "tl_uncons" tl_uncons,
1001+ testProperty "t_unsnoc" t_unsnoc,
1002+ testProperty "tl_unsnoc" tl_unsnoc,
9701003 testProperty "s_head" s_head,
9711004 testProperty "sf_head" sf_head,
9721005 testProperty "t_head" t_head,
@@ -1030,7 +1063,9 @@ tests =
10301063 testProperty "tl_toLower_lower" tl_toLower_lower,
10311064 testProperty "t_toUpper_length" t_toUpper_length,
10321065 testProperty "t_toUpper_upper" t_toUpper_upper,
1033− testProperty "tl_toUpper_upper" tl_toUpper_upper
1066+ testProperty "tl_toUpper_upper" tl_toUpper_upper,
1067+ testProperty "t_toTitle_title" t_toTitle_title,
1068+ testProperty "t_toTitle_1stNotLower" t_toTitle_1stNotLower
10341069 ],
10351070
10361071 testGroup "justification" [
@@ -1147,7 +1182,9 @@ tests =
11471182 testProperty "t_takeWhile" t_takeWhile,
11481183 testProperty "tl_takeWhile" tl_takeWhile,
11491184 testProperty "t_takeWhileEnd" t_takeWhileEnd,
1185+ testProperty "t_takeWhileEnd_null" t_takeWhileEnd_null,
11501186 testProperty "tl_takeWhileEnd" tl_takeWhileEnd,
1187+ testProperty "tl_takeWhileEnd_null" tl_takeWhileEnd_null,
11511188 testProperty "sf_dropWhile" sf_dropWhile,
11521189 testProperty "s_dropWhile" s_dropWhile,
11531190 testProperty "s_dropWhile_s" s_dropWhile_s,