Skip to content
This repository has been archived by the owner on Mar 31, 2024. It is now read-only.

Commit

Permalink
Test for unicode error in magic methods (Python 2)
Browse files Browse the repository at this point in the history
  • Loading branch information
Andrew Savchyn committed Dec 13, 2015
1 parent 2bd5aef commit e101c56
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions test_clint.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,25 +18,25 @@ def tearDown(self):
pass

class ColoredStringTestCase(unittest.TestCase):

def setUp(self):
from clint.textui.colored import ColoredString

def tearDown(self):
pass

def test_split(self):
from clint.textui.colored import ColoredString
new_str = ColoredString('red', "hello world")
output = new_str.split()
assert output[0].s == "hello"

def test_find(self):
from clint.textui.colored import ColoredString
new_str = ColoredString('blue', "hello world")
output = new_str.find('h')
self.assertEqual(output, 0)

def test_replace(self):
from clint.textui.colored import ColoredString
new_str = ColoredString('green', "hello world")
Expand All @@ -49,14 +49,19 @@ def test_py2_bytes_not_mangled(self):
new_str = ColoredString('RED', '\xe4')
assert '\xe4' in str(new_str)
from clint.textui import puts
puts(new_str)

def test_clint_force_color_env_var(self):
from clint.textui.colored import ColoredString
os.environ['CLINT_FORCE_COLOR'] = "1"
new_str = ColoredString('RED', 'hello world')
assert new_str.always_color == True

def test_clint_unicode_radd(self):
from clint.textui.colored import ColoredString
inp_str = u'hello \u263A'
new_str = u'' + ColoredString('RED', inp_str)
assert inp_str.encode('utf-8') in new_str


class TextuiFormatterTestCase(unittest.TestCase):

Expand Down

0 comments on commit e101c56

Please sign in to comment.