def test_strip_codes(self): """ strip_codes() should strip all color and reset codes. """ self.assertEqual( 'hello world', strip_codes( Colr('hello world', fore='green', back='blue', style='bright') ), msg=test_msg('Failed to strip codes from Colr string.')) self.assertEqual( 'hello world', strip_codes( color('hello world', fore='red', back='blue', style='bright') ), msg=test_msg('Failed to strip codes from color string.')) self.assertEqual( 'hello world', strip_codes( Colr().red().bggreen().bright('hello world') ), msg=test_msg('Failed to strip codes from chained Colr string.')) self.assertEqual( 'hello world', strip_codes( Colr('hello world').rainbow() ), msg=test_msg('Failed to strip codes from Colr.rainbow string.'))
def ellipsize(self, line: str, offset: int = 0): if not isinstance(line, str): line = repr(line) length = self.width - 5 - offset if len(strip_codes(line)) < length: return line return '%s(...)' % line[:length]
def test_strip_codes(self): """ strip_codes() should strip all color and reset codes. """ s = '\n'.join(('This is a test of strip_codes.', 'There should be none after stripping.')) colrvals = { 'Colr': Colr(s, fore='green', back='blue', style='bright'), 'color func': color(s, fore='red', back='blue', style='bright'), 'chained': Colr().red().bggreen().bright(s), 'chained extended': Colr().f_55().b_55().bright(s), 'chained rgb': Colr().rgb(25, 25, 25).b_rgb(55, 55, 55).bright(s), 'Colr.rainbow': Colr(s).rainbow(), 'Colr.rainbow rgb': Colr(s).rainbow(rgb_mode=True), } for desc, colrval in colrvals.items(): self.assertCallEqual( s, strip_codes(colrval), func=strip_codes, args=[colrval], msg='Failed to strip codes from {} string.'.format(desc), )
def test_strip_codes(self): """ strip_codes() should strip all color and reset codes. """ s = '\n'.join(('This is a test of strip_codes.', 'There should be none after stripping.')) self.assertEqual( s, strip_codes(Colr(s, fore='green', back='blue', style='bright')), msg=call_msg('Failed to strip codes from Colr string.')) self.assertEqual( s, strip_codes(color(s, fore='red', back='blue', style='bright')), msg=call_msg('Failed to strip codes from color string.')) self.assertEqual( s, strip_codes(Colr().red().bggreen().bright(s)), msg=call_msg('Failed to strip codes from chained Colr string.')) self.assertEqual( s, strip_codes(Colr().f_55().b_55().bright(s)), msg=call_msg('Failed to strip codes from extended Colr string.')) self.assertEqual( s, strip_codes(Colr().rgb(25, 25, 25).b_rgb(55, 55, 55).bright(s)), msg=call_msg('Failed to strip codes from rgb Colr string.')) self.assertEqual( s, strip_codes(Colr(s).rainbow()), msg=call_msg('Failed to strip codes from Colr.rainbow string.')) self.assertEqual( s, strip_codes(Colr(s).rainbow(rgb_mode=True)), msg=call_msg( 'Failed to strip codes from Colr.rainbow rgb string.'))