update tests

jrnl/main.py

@@ -1,7 +1,13 @@
 # Copyright © 2012-2023 jrnl contributors
 # License: https://www.gnu.org/licenses/gpl-3.0.html
 from ruamel import yaml
+import sys
 
-def run():
-    my_dict = dict(a='a' * 80, b=dict(c=1, d=2))
+def run(num = None):
+    if num is None:
+        num = sys.argv[1]
+
+    my_str = "a" * int(num)
+    my_dict = dict(a=my_str, b=dict(c=1, d=2))
     return yaml.dump(my_dict)
+

pyproject.toml

@@ -33,6 +33,6 @@ isolated_build = True
 deps =
   pytest >= 6.2
 
-commands = pytest -vvv {posargs}
+commands = pytest -v {posargs}
 passenv = HOME
 """

tests/test_run.py

@@ -4,9 +4,12 @@
 from jrnl.main import run
 
 
-def test_whatever():
-    expected_output = """
-        a: aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
-        b: {c: 1, d: 2}
-    """.strip()
-    assert run() == expected_output
+def test_passes():
+    num = 80
+    assert run(num) == "a: " + ("a" * num) +  "\nb: {c: 1, d: 2}\n"
+
+
+def test_fails():
+    num = 81
+    assert run(num) == "a: " + ("a" * num) +  "\nb: {c: 1, d: 2}\n"
+