From: Tim Peters <tim.peters@gmail.com>
Date: Fri, 27 Aug 2004 05:36:07 +0000 (+0000)
Subject: test_bug1001011():  Verify that
X-Git-Tag: v2.4a3~86
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=108f13751999f63c20fa9067f3fc0feb6a87718e;p=python

test_bug1001011():  Verify that

    s.join([t]) is t

for (s, t) in (str, str), (unicode, unicode), and (str, unicode).
For (unicode, str), verify that it's *not* t (the result is promoted
to unicode instead).  Also verify that when t is a subclass of str or
unicode that "the right thing" happens.
---

diff --git a/Lib/test/string_tests.py b/Lib/test/string_tests.py
index 9cf6b9e1cf..4335965b36 100644
--- a/Lib/test/string_tests.py
+++ b/Lib/test/string_tests.py
@@ -699,14 +699,13 @@ class MixinStrUserStringTest:
 
 
 class MixinStrUnicodeTest:
-    # Additional tests that only work with
-    # str and unicode
+    # Additional tests that only work with str and unicode.
 
     def test_bug1001011(self):
         # Make sure join returns a NEW object for single item sequences
-        # involving a subclass
-        # Make sure that it is of the appropriate type
-        # Check the optimisation still occurs for standard objects
+        # involving a subclass.
+        # Make sure that it is of the appropriate type.
+        # Check the optimisation still occurs for standard objects.
         t = self.type2test
         class subclass(t):
             pass
@@ -714,3 +713,32 @@ class MixinStrUnicodeTest:
         s2 = t().join([s1])
         self.assert_(s1 is not s2)
         self.assert_(type(s2) is t)
+
+        s1 = t("abcd")
+        s2 = t().join([s1])
+        self.assert_(s1 is s2)
+
+        # Should also test mixed-type join.
+        if t is unicode:
+            s1 = subclass("abcd")
+            s2 = "".join([s1])
+            self.assert_(s1 is not s2)
+            self.assert_(type(s2) is t)
+
+            s1 = t("abcd")
+            s2 = "".join([s1])
+            self.assert_(s1 is s2)
+
+        elif t is str:
+            s1 = subclass("abcd")
+            s2 = u"".join([s1])
+            self.assert_(s1 is not s2)
+            self.assert_(type(s2) is unicode) # promotes!
+
+            s1 = t("abcd")
+            s2 = u"".join([s1])
+            self.assert_(s1 is not s2)
+            self.assert_(type(s2) is unicode) # promotes!
+
+        else:
+            self.fail("unexpected type for MixinStrUnicodeTest %r" % t)