Fixed issues with transformation functions on ANSIString.

2014-04-07 11:12:59 -05:00 · 2014-04-07 11:12:59 -05:00 · 834b039a0d
commit 834b039a0d
parent 7fd47eb386
2 changed files with 12 additions and 1 deletions
--- a/src/utils/tests.py
+++ b/src/utils/tests.py
@ -110,3 +110,12 @@ class ANSIStringTestCase(TestCase):
        ANSI codes.
        """
        self.assertEqual(len(ANSIString('{gTest{n')), 4)
+
+    def test_capitalize(self):
+        """
+        Make sure that capitalization works. This is the simplest of the
+        _transform functions.
+        """
+        target = ANSIString('{gtest{n')
+        result = u'\x1b[1m\x1b[32mTest\x1b[0m'
+        self.checker(target.capitalize(), result, u'Test')