From 3a218b9a153cb074e1953a4910451a12f5fbbd76 Mon Sep 17 00:00:00 2001 From: Matthew Peveler Date: Wed, 9 Oct 2019 15:38:35 -0300 Subject: [PATCH] fix: add reset_replace_colors to __all__ --- crayons.py | 5 +++-- test_crayons.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) 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!'))