fix style

Textualize · Mar 2, 2020 · 9a98dec · 9a98dec
1 parent a8736e2
commit 9a98dec
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/rich/traceback.py b/rich/traceback.py
@@ -39,7 +39,7 @@ def install(
     Args:
         width (Optional[int], optional): Width (in characters) of traceback. Defaults to 100.
         extra_lines (int, optional): Extra lines of code. Defaults to 3.
-        theme (str, optional) Pygments theme to use in traceback. Defaults to ``None`` which will pick
+        theme (Optional[str], optional) Pygments theme to use in traceback. Defaults to ``None`` which will pick
             a theme appropriate for the platform. 
 
     """

diff --git a/tests/test_console.py b/tests/test_console.py
@@ -25,14 +25,14 @@ def test_style_context():
     console = Console()
 
     with StyleContext(console, None):
-        assert console.current_style == Style()
+        assert console._current_style == Style()
 
     with StyleContext(console, "bold"):
-        assert console.current_style == Style.parse("bold")
+        assert console._current_style == Style.parse("bold")
         with StyleContext(console, "red"):
-            assert console.current_style == Style.parse("bold red")
-        assert console.current_style == Style.parse("bold")
-    assert console.current_style == Style()
+            assert console._current_style == Style.parse("bold red")
+        assert console._current_style == Style.parse("bold")
+    assert console._current_style == Style()
 
 
 def test_init():