str: proper titlecase support

OddCoincidence · OddCoincidence · commit 79623f4e7a3e · 2019-04-14T12:23:09.000-07:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/vm/Cargo.toml b/vm/Cargo.toml
@@ -23,6 +23,7 @@ regex = "1"
 rustc_version_runtime = "0.1.*"
 statrs = "0.10.0"
 caseless = "0.2.1"
+unicode-casing = { git = "https://github.com/OddCoincidence/unicode-casing" }
 unicode-segmentation = "1.2.1"
 lazy_static = "^1.0.1"
 lexical = "2.0.0"
diff --git a/vm/src/obj/objstr.rs b/vm/src/obj/objstr.rs
@@ -5,6 +5,7 @@ use std::str::FromStr;
 use std::string::ToString;
 
 use num_traits::ToPrimitive;
+use unicode_casing::CharExt;
 use unicode_segmentation::UnicodeSegmentation;
 
 use crate::format::{FormatParseError, FormatPart, FormatString};
@@ -413,12 +414,12 @@ impl PyString {
         for c in self.value.chars() {
             if c.is_lowercase() {
                 if !previous_is_cased {
-                    title.extend(c.to_uppercase());
+                    title.extend(c.to_titlecase());
                 } else {
                     title.push(c);
                 }
                 previous_is_cased = true;
-            } else if c.is_uppercase() {
+            } else if c.is_uppercase() || c.is_titlecase() {
                 if previous_is_cased {
                     title.extend(c.to_lowercase());
                 } else {
@@ -652,7 +653,7 @@ impl PyString {
         let mut cased = false;
         let mut previous_is_cased = false;
         for c in self.value.chars() {
-            if c.is_uppercase() {
+            if c.is_uppercase() || c.is_titlecase() {
                 if previous_is_cased {
                     return false;
                 }
@@ -1050,6 +1051,7 @@ mod tests {
             ("Format,This-As*Title;String", "fOrMaT,thIs-aS*titLe;String"),
             ("Getint", "getInt"),
             ("Greek Ωppercases ...", "greek ωppercases ..."),
+            ("Greek ῼitlecases ...", "greek ῳitlecases ..."),
         ];
         for (title, input) in tests {
             assert_eq!(PyString::from(input).title(&vm).as_str(), title);
@@ -1066,6 +1068,7 @@ mod tests {
             "A\nTitlecased Line",
             "A Titlecased, Line",
             "Greek Ωppercases ...",
+            "Greek ῼitlecases ...",
         ];
 
         for s in pos {
