@@ -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,25 @@ 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)
545+t_takeWhile_null n c
546+ = T.null $ T.takeWhile (==c) (T.replicate n (T.singleton c))
518547tl_takeWhile p = L.takeWhile p `eqP` (unpackS . TL.takeWhile p)
548+tl_takeWhile_null n c
549+ = TL.null $
550+ TL.takeWhile (==c) (TL.replicate n (TL.singleton c))
519551t_takeWhileEnd p = (L.reverse . L.takeWhile p . L.reverse) `eqP`
520552 (unpackS . T.takeWhileEnd p)
553+t_takeWhileEnd_null t = forAll noMatch $ \(c,d) -> T.null $
554+ T.takeWhileEnd (==d) (T.snoc t c)
521555tl_takeWhileEnd p = (L.reverse . L.takeWhile p . L.reverse) `eqP`
522556 (unpackS . TL.takeWhileEnd p)
557+tl_takeWhileEnd_null t = forAll noMatch $ \(c,d) -> TL.null $
558+ TL.takeWhileEnd (==d) (TL.snoc t c)
523559s_dropWhile p = L.dropWhile p `eqP` (unpackS . S.dropWhile p)
524560s_dropWhile_s p = L.dropWhile p `eqP` (unpackS . S.unstream . S.dropWhile p)
525561sf_dropWhile q p = (L.dropWhile p . L.filter q) `eqP`
@@ -720,7 +756,7 @@ tl_indices (NotEmpty s) = lazyIndices s `eq` S.indices s
720756 conc = T.concat . TL.toChunks
721757t_indices_occurs = unsquare $ \(NotEmpty t) ts ->
722758 let s = T.intercalate t ts
723− in Slow.indices t s == indices t s
759+ in Slow.indices t s === indices t s
724760
725761-- Bit shifts.
726762shiftL w = forAll (choose (0,width-1)) $ \k -> Bits.shiftL w k == U.shiftL w k
@@ -812,18 +848,18 @@ tb_formatRealFloat_double (a::Double) = tb_formatRealFloat a
812848-- Reading.
813849
814850t_decimal (n::Int) s =
815− T.signed T.decimal (T.pack (show n) `T.append` t) == Right (n,t)
851+ T.signed T.decimal (T.pack (show n) `T.append` t) === Right (n,t)
816852 where t = T.dropWhile isDigit s
817853tl_decimal (n::Int) s =
818− TL.signed TL.decimal (TL.pack (show n) `TL.append` t) == Right (n,t)
854+ TL.signed TL.decimal (TL.pack (show n) `TL.append` t) === Right (n,t)
819855 where t = TL.dropWhile isDigit s
820856t_hexadecimal m s ox =
821− T.hexadecimal (T.concat [p, T.pack (showHex n ""), t]) == Right (n,t)
857+ T.hexadecimal (T.concat [p, T.pack (showHex n ""), t]) === Right (n,t)
822858 where t = T.dropWhile isHexDigit s
823859 p = if ox then "0x" else ""
824860 n = getPositive m :: Int
825861tl_hexadecimal m s ox =
826− TL.hexadecimal (TL.concat [p, TL.pack (showHex n ""), t]) == Right (n,t)
862+ TL.hexadecimal (TL.concat [p, TL.pack (showHex n ""), t]) === Right (n,t)
827863 where t = TL.dropWhile isHexDigit s
828864 p = if ox then "0x" else ""
829865 n = getPositive m :: Int
@@ -867,11 +903,11 @@ tl_write_read_line e m b t = write_read head TL.filter TL.hPutStrLn
867903
868904t_dropWord16 m t = dropWord16 m t `T.isSuffixOf` t
869905t_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
906+t_take_drop_16 m t = T.append (takeWord16 n t) (dropWord16 n t) === t
871907 where n = small m
872908t_use_from t = monadicIO $ assert . (==t) =<< run (useAsPtr t fromPtr)
873909
874−t_copy t = T.copy t == t
910+t_copy t = T.copy t === t
875911
876912-- Regression tests.
877913s_filter_eq s = S.filter p t == S.streamList (filter p s)
@@ -967,6 +1003,8 @@ tests =
9671003 testProperty "sf_uncons" sf_uncons,
9681004 testProperty "t_uncons" t_uncons,
9691005 testProperty "tl_uncons" tl_uncons,
1006+ testProperty "t_unsnoc" t_unsnoc,
1007+ testProperty "tl_unsnoc" tl_unsnoc,
9701008 testProperty "s_head" s_head,
9711009 testProperty "sf_head" sf_head,
9721010 testProperty "t_head" t_head,
@@ -1030,7 +1068,9 @@ tests =
10301068 testProperty "tl_toLower_lower" tl_toLower_lower,
10311069 testProperty "t_toUpper_length" t_toUpper_length,
10321070 testProperty "t_toUpper_upper" t_toUpper_upper,
1033− testProperty "tl_toUpper_upper" tl_toUpper_upper
1071+ testProperty "tl_toUpper_upper" tl_toUpper_upper,
1072+ testProperty "t_toTitle_title" t_toTitle_title,
1073+ testProperty "t_toTitle_1stNotLower" t_toTitle_1stNotLower
10341074 ],
10351075
10361076 testGroup "justification" [
@@ -1145,9 +1185,13 @@ tests =
11451185 testProperty "s_takeWhile_s" s_takeWhile_s,
11461186 testProperty "sf_takeWhile" sf_takeWhile,
11471187 testProperty "t_takeWhile" t_takeWhile,
1188+ testProperty "t_takeWhile_null" t_takeWhile_null,
11481189 testProperty "tl_takeWhile" tl_takeWhile,
1190+ testProperty "tl_takeWhile_null" tl_takeWhile_null,
11491191 testProperty "t_takeWhileEnd" t_takeWhileEnd,
1192+ testProperty "t_takeWhileEnd_null" t_takeWhileEnd_null,
11501193 testProperty "tl_takeWhileEnd" tl_takeWhileEnd,
1194+ testProperty "tl_takeWhileEnd_null" tl_takeWhileEnd_null,
11511195 testProperty "sf_dropWhile" sf_dropWhile,
11521196 testProperty "s_dropWhile" s_dropWhile,
11531197 testProperty "s_dropWhile_s" s_dropWhile_s,