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

10
jrnl/main.py
View file

@ -1,13 +1,17 @@
# Copyright © 2012-2023 jrnl contributors
# License: https://www.gnu.org/licenses/gpl-3.0.html
from ruamel import yaml
from ruamel.yaml import YAML
import sys
from io import StringIO
def run(num = None):
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)
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():
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():
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"