From b0d48558606fabfb582358077c3c909c2bbb63e1 Mon Sep 17 00:00:00 2001
From: David Luevano Alvarado <david@luevano.xyz>
Date: Thu, 16 Feb 2023 02:22:51 -0600
Subject: add type hints

---
 tests/test_pymdvar.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

(limited to 'tests')

diff --git a/tests/test_pymdvar.py b/tests/test_pymdvar.py
index 69ffe56..91a97d0 100644
--- a/tests/test_pymdvar.py
+++ b/tests/test_pymdvar.py
@@ -4,10 +4,8 @@ from pymdvar import VariableExtension
 
 
 def test_empty_input():
-    in_str = ''
-    out_str = markdown(in_str, extensions=[VariableExtension()])
-    print(in_str)
-    print(out_str)
+    in_str: str = ''
+    out_str: str = markdown(in_str, extensions=[VariableExtension()])
     assert in_str == out_str
 
 
@@ -20,7 +18,7 @@ def test_empty_input():
     ('foo **$test}** bar', '<p>foo <strong>$test}</strong> bar</p>'),
 ])
 def test_non_replacements(in_str, exp_str):
-    out_str = markdown(in_str, extensions=[VariableExtension()])
+    out_str: str = markdown(in_str, extensions=[VariableExtension()])
     assert out_str == exp_str
 
 
@@ -32,6 +30,6 @@ def test_non_replacements(in_str, exp_str):
     ('foo ![image](${test}/a.jpg) bar', '<p>foo <img alt="image" src="value/a.jpg" /> bar</p>'),
 ])
 def test_simple_replacements(in_str, exp_str):
-    out_str = markdown(in_str, extensions=[VariableExtension(variables={'test':'value'})])
+    out_str: str = markdown(in_str, extensions=[VariableExtension(variables={'test':'value'})])
     assert out_str == exp_str
 
-- 
cgit v1.2.3-70-g09d2