def test_colorizer_changes_colorizable_types(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), }) self.assertEqual( ColorizedObject(Mark(value, 'a'), ('[', ']')), colorizer.colorize(Mark(value, 'a')), )
def test_colorizer_changes_colorizable_types(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), }) self.assertEqual( ColorizedObject(Mark(value, 'a'), ('[', ']')), colorizer.colorize(Mark(value, 'a')), )
def test_colorizer_changes_colorizable_types_with_tags(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual( ColorizedObject(Mark(value, ['a', 'b']), ('[<', '>]')), colorizer.colorize(Mark(value, ['a', 'b'])), )
def test_colorizer_get_color_pair_not_found_with_default(self): colorizer = Colorizer( { 'a': ('[', ']'), 'b': ('<', '>'), }, default_color_tag='b', ) self.assertEqual(('<', '>'), colorizer.get_color_pair(color_tag=['c']))
def test_colorizer_get_color_pair_found_with_context(self): colorizer = Colorizer({ 'a': ('[', ']'), 'b': ('<', '>'), }, ) self.assertEqual( ('><[', ']><'), colorizer.get_color_pair(color_tag=['a'], context_color_tag='b'), )
def test_colorizer_changes_colorizable_types_with_tags(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual( ColorizedObject(Mark(value, ['a', 'b']), ('[<', '>]')), colorizer.colorize(Mark(value, ['a', 'b'])), )
def test_colorizer_get_color_pair_found_double(self): colorizer = Colorizer({ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual( ('[<', '>]'), colorizer.get_color_pair(color_tag=['a', 'b']), )
def test_colorizer_doesnt_colorize_with_unknown_color_tag(self): colorizer = Colorizer(color_map={ 'my_tag': ('START_MARK', 'STOP_MARK'), }, ) result = colorizer.colorize(Mark('hello', color_tag='my_unknown_tag')) self.assertEqual( ColorizedObject(Mark('hello', 'my_unknown_tag'), ('', '')), result, )
def test_colorizer_get_color_pair_not_found_with_default(self): colorizer = Colorizer( { 'a': ('[', ']'), 'b': ('<', '>'), }, default_color_tag='b', ) self.assertEqual(('<', '>'), colorizer.get_color_pair(color_tag=['c']))
def test_colorizer_get_color_pair_found_double(self): colorizer = Colorizer({ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual( ('[<', '>]'), colorizer.get_color_pair(color_tag=['a', 'b']), )
def test_colorizer_changes_colorizable_types_with_context(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual( ColorizedObject(Mark(value, 'a'), ('><[', ']><')), colorizer.colorize(Mark(value, 'a'), context_color_tag='b'), )
def test_colorizer_changes_colorizable_types_with_context(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual( ColorizedObject(Mark(value, 'a'), ('><[', ']><')), colorizer.colorize(Mark(value, 'a'), context_color_tag='b'), )
def test_colorizer_get_color_pair_found_with_context(self): colorizer = Colorizer( { 'a': ('[', ']'), 'b': ('<', '>'), }, ) self.assertEqual( ('><[', ']><'), colorizer.get_color_pair(color_tag=['a'], context_color_tag='b'), )
def test_colorizer_doesnt_colorize_with_unknown_color_tag(self): colorizer = Colorizer( color_map={ 'my_tag': ('START_MARK', 'STOP_MARK'), }, ) result = colorizer.colorize(Mark('hello', color_tag='my_unknown_tag')) self.assertEqual( ColorizedObject(Mark('hello', 'my_unknown_tag'), ('', '')), result, )
def test_colorizer_changes_colorizable_types_with_tags_and_context( self, _, value, ): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('(', ')'), 'c': ('<', '>'), }) self.assertEqual( ColorizedObject(Mark(value, ['a', 'b']), ('><[(', ')]><')), colorizer.colorize(Mark(value, ['a', 'b']), context_color_tag='c'), )
def test_colorizer_get_color_pair_found_with_list_context(self): colorizer = Colorizer( { 'a': ('[', ']'), 'b': ('<', '>'), 'c': ('(', ')'), }, ) self.assertEqual( (')><([', '])><('), colorizer.get_color_pair( color_tag=['a'], context_color_tag=['b', 'c'], ), )
def test_colorizer_changes_colorizable_types_with_tags_and_context( self, _, value, ): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('(', ')'), 'c': ('<', '>'), }) self.assertEqual( ColorizedObject(Mark(value, ['a', 'b']), ('><[(', ')]><')), colorizer.colorize(Mark(value, ['a', 'b']), context_color_tag='c'), )
def test_colorize_message(self): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('(', ')'), }) message = '{}-{}_{a}~{b}' args = [42, Mark(42, ['a', 'b'])] kwargs = { 'a': 0, 'b': Mark(0, ['b', 'a']), } self.assertEqual( '42-[(42)]_0~([0])', colorizer.colorize_message(message, *args, **kwargs), )
def test_colorize_message(self): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('(', ')'), }) message = '{0}-{1}_{a}~{b}' args = [42, Mark(42, ['a', 'b'])] kwargs = { 'a': 0, 'b': Mark(0, ['b', 'a']), } self.assertEqual( '42-[(42)]_0~([0])', colorizer.colorize_message(message, *args, **kwargs), )
def test_colorizer_get_color_pair_found_with_list_context(self): colorizer = Colorizer( { 'a': ('[', ']'), 'b': ('<', '>'), 'c': ('(', ')'), }, ) self.assertEqual( (')><([', '])><('), colorizer.get_color_pair( color_tag=['a'], context_color_tag=['b', 'c'], ), )
def test_colorize_message_with_context(self): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('(', ')'), 'c': ('<', '>'), }) message = Mark('{0}-{1}_{a}~{b}', 'c') args = [42, Mark(42, ['a', 'b'])] kwargs = { 'a': 0, 'b': Mark(0, ['b', 'a']), } self.assertEqual( '<42-><[(42)]><_0~><([0])><>', colorizer.colorize_message(message, *args, **kwargs), )
def test_colorize_message_with_context(self): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('(', ')'), 'c': ('<', '>'), }) message = Mark('{}-{}_{a}~{b}', 'c') args = [42, Mark(42, ['a', 'b'])] kwargs = { 'a': 0, 'b': Mark(0, ['b', 'a']), } self.assertEqual( '<42-><[(42)]><_0~><([0])><>', colorizer.colorize_message(message, *args, **kwargs), )
def test_colorizer_colorizes_with_unknown_color_tag_and_default(self): colorizer = Colorizer( color_map={ 'my_tag': ('START_MARK', 'STOP_MARK'), 'default': ('START_DEFAULT_MARK', 'STOP_DEFAULT_MARK') }, default_color_tag='default', ) result = colorizer.colorize(Mark('hello', color_tag='my_unknown_tag')) self.assertEqual( ColorizedObject( Mark( 'hello', 'my_unknown_tag', ), ( 'START_DEFAULT_MARK', 'STOP_DEFAULT_MARK', ), ), result, )
def test_colorizer_colorizes_with_unknown_color_tag_and_default(self): colorizer = Colorizer( color_map={ 'my_tag': ('START_MARK', 'STOP_MARK'), 'default': ('START_DEFAULT_MARK', 'STOP_DEFAULT_MARK') }, default_color_tag='default', ) result = colorizer.colorize(Mark('hello', color_tag='my_unknown_tag')) self.assertEqual( ColorizedObject( Mark( 'hello', 'my_unknown_tag', ), ( 'START_DEFAULT_MARK', 'STOP_DEFAULT_MARK', ), ), result, )
def __init__(self, stream, colorizer=Colorizer()): """ Initialize the :class:`StreamDisplay`. :param stream: The stream to be attached too. """ super(StreamDisplay, self).__init__() self.colorizer = colorizer self.output_map = {} if stream_has_color_support(stream): self.stream = AnsiToWin32(stream).stream else: self.stream = stream # Python 3 differentiates binary streams. if hasattr(stream, 'buffer'): self.binary_stream = stream.buffer else: self.binary_stream = stream
def test_colorizer_get_color_pair_not_found(self): colorizer = Colorizer({}) self.assertEqual(('', ''), colorizer.get_color_pair(color_tag=['a']))
def test_colorizer_converts_unknown_types(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual(ColorizedObject(value), colorizer.colorize(value))
def test_colorizer_get_color_pair_found(self): colorizer = Colorizer({ 'a': ('[', ']'), }) self.assertEqual(('[', ']'), colorizer.get_color_pair(color_tag=['a']))
def test_colorizer_converts_unknown_types(self, _, value): colorizer = Colorizer(color_map={ 'a': ('[', ']'), 'b': ('<', '>'), }) self.assertEqual(ColorizedObject(value), colorizer.colorize(value))
def test_colorizer_get_color_pair_not_found(self): colorizer = Colorizer({}) self.assertEqual(('', ''), colorizer.get_color_pair(color_tag=['a']))
def test_colorizer_get_color_pair_found(self): colorizer = Colorizer({ 'a': ('[', ']'), }) self.assertEqual(('[', ']'), colorizer.get_color_pair(color_tag=['a']))