fz0x1/jrnl
1
0
Fork 0
mirror of https://github.com/jrnl-org/jrnl.git synced 2025-05-10 08:38:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update tests

Browse source
This commit is contained in:
Jonathan Wren 2023-05-06 16:28:33 -07:00
parent 7b5d3fdaf9
commit d40b013bcd
No known key found for this signature in database
2 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
jrnl/main.py
View file

@ -1,7 +1,9 @@
# Copyright © 2012-2023 jrnl contributors # Copyright © 2012-2023 jrnl contributors
# License: https://www.gnu.org/licenses/gpl-3.0.html # License: https://www.gnu.org/licenses/gpl-3.0.html
from ruamel import yaml from ruamel.yaml import YAML
import sys import sys
from io import StringIO
def run(num=None): def run(num=None):
if num is None: if num is None:
@ -9,5 +11,7 @@ def run(num = None):
my_str = "a" * int(num) my_str = "a" * int(num)
my_dict = dict(a=my_str, b=dict(c=1, d=2)) my_dict = dict(a=my_str, b=dict(c=1, d=2))
return yaml.dump(my_dict)
output = StringIO()
YAML().dump(my_dict, output)
return output.getvalue()

4
tests/test_run.py
View file

@ -6,10 +6,10 @@ from jrnl.main import run
def test_passes(): def test_passes():
num = 80 num = 80
assert run(num) == "a: " + ("a" * num) + "\nb: {c: 1, d: 2}\n" assert run(num) == "a: " + ("a" * num) + "\nb:\n c: 1\n d: 2\n"
def test_fails(): def test_fails():
num = 81 num = 81
assert run(num) == "a: " + ("a" * num) + "\nb: {c: 1, d: 2}\n" assert run(num) == "a: " + ("a" * num) + "\nb:\n c: 1\n d: 2\n"