diff --git a/crayons.py b/crayons.py index 021edad..cd79a74 100644 --- a/crayons.py +++ b/crayons.py @@ -17,7 +17,8 @@ COLORS = ( ) __all__ = COLORS + ( - 'normal', 'clean', 'disable', 'enable', 'random', 'replace_colors' + 'normal', 'clean', 'disable', 'enable', 'random', + 'replace_colors', 'reset_replace_colors' ) colorama.init() @@ -169,7 +170,7 @@ def replace_colors(replace_dict): REPLACE_COLORS = {k.upper(): v.upper() for k, v in replace_dict.items()} -def reset_replace(): +def reset_replace_colors(): """Reset any set replace colors.""" global REPLACE_COLORS diff --git a/test_crayons.py b/test_crayons.py index 35d3abf..fd07d0f 100755 --- a/test_crayons.py +++ b/test_crayons.py @@ -44,5 +44,5 @@ print(clean('{} clean {}'.format(red('red'), blue('blue')))) # NOQA crayons.replace_colors({'magenta': 'blue'}) print(crayons.magenta('this is blue!')) -crayons.reset_replace() +crayons.reset_replace_colors() print(crayons.magenta('this is magenta again!'))