X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=tests%2Fstring_utils_test.py;h=51848946b68c3b661cb21d6b7d38c3eaad15e5ef;hb=aca44b9aff1ea315c18b80a1c543f4e9b79fc9b0;hp=0472daaccaf9a525794df24e79c8ae5f923898f0;hpb=3bc4daf1edc121cd633429187392227f2fa61885;p=python_utils.git

diff --git a/tests/string_utils_test.py b/tests/string_utils_test.py
index 0472daa..5184894 100755
--- a/tests/string_utils_test.py
+++ b/tests/string_utils_test.py
@@ -2,16 +2,14 @@
 
 import unittest
 
-from ansi import fg, bg, reset
 import bootstrap
 import string_utils as su
-
 import unittest_utils as uu
+from ansi import bg, fg, reset
 
 
 @uu.check_all_methods_for_perf_regressions()
 class TestStringUtils(unittest.TestCase):
-
     def test_is_none_or_empty(self):
         self.assertTrue(su.is_none_or_empty(None))
         self.assertTrue(su.is_none_or_empty(""))
@@ -130,7 +128,7 @@ class TestStringUtils(unittest.TestCase):
         s = f' {fg("red")}  this is a test  {bg("white")}  this is a test  {reset()}  '
         self.assertEqual(
             su.strip_escape_sequences(s),
-            '   this is a test    this is a test    '
+            '   this is a test    this is a test    ',
         )
         s = ' this is another test '
         self.assertEqual(su.strip_escape_sequences(s), s)
@@ -180,6 +178,12 @@ class TestStringUtils(unittest.TestCase):
         self.assertFalse(su.is_snake_case('thisIsATest'))
         self.assertTrue(su.is_snake_case('this_is_a_test'))
 
+    def test_sprintf_context(self):
+        with su.SprintfStdout() as buf:
+            print("This is a test.")
+            print("This is another one.")
+        self.assertEqual('This is a test.\nThis is another one.\n', buf())
+
 
 if __name__ == '__main__':
     bootstrap.initialize(unittest.main)()