fix the color management with rich

cloudmesh · Jan 8, 2024 · aa586f0 · aa586f0
1 parent b338b1c
commit aa586f0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/src/cloudmesh/common/console.py b/src/cloudmesh/common/console.py
@@ -362,7 +362,7 @@ def cprint(color="BLUE", prefix="", message=""):
         message = message or ""
         prefix = prefix or ""
         #RichConsole.print(prefix + message, style=color.lower())
-        c = Console.theme[color]
+        c = Console.theme[color.upper()]
         RichConsole.print(prefix + message, style=c)
 
 

diff --git a/tests/test_shell.py b/tests/test_shell.py
@@ -109,7 +109,7 @@ def test_open(self):
         if os_is_windows() and github_action:
             pytest.skip('not supported')
         Benchmark.Start()
-        filename = 'cloudmesh/common/console.py'
+        filename = 'src/cloudmesh/common/console.py'
         Shell.open(filename)
         Benchmark.Stop()
         assert True  # has to be a visual test!

diff --git a/tests/test_shell_tests.py b/tests/test_shell_tests.py
@@ -162,9 +162,8 @@ def test_shell_head(self):
         file = path_expand('requirements.txt')
         r = Shell.head(file)
         Benchmark.Stop()
-        assert 'tqdm' in r
-        assert 'rich' in r
-        assert 'tabulate' in r
+        assert 'humanize' in r
+        assert 'oyaml' in r
         r = Shell.head('requirements.txt', lines=1)
         assert 'cloudmesh-sys' not in r