Skip to content

Commit 06465e0

Browse files
committed
Added tests for save_values
1 parent d7d38bf commit 06465e0

File tree

1 file changed

+43
-1
lines changed

1 file changed

+43
-1
lines changed

test/test_cli.py

+43-1
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,49 @@ def test_save_data():
108108

109109

110110
def test_save_values():
111-
pass # TODO
111+
values = {
112+
'date_thresh': date(2020, 8, 8),
113+
114+
'offset_mean': 1.3905680578542898,
115+
'offset_stdev': 1.0707342960061585,
116+
'offset_upper': 3.532036649866607,
117+
'offset_lower': -0.7509005341580273,
118+
119+
'sugg_src': 857.5727674560864,
120+
'sugg_dst': 6.860582139648691,
121+
}
122+
123+
txt = textwrap.dedent('''\
124+
date_thresh=2020-08-08
125+
offset_mean=1.3905680578542898
126+
offset_stdev=1.0707342960061585
127+
offset_upper=3.532036649866607
128+
offset_lower=-0.7509005341580273
129+
sugg_src=857.5727674560864
130+
sugg_dst=6.860582139648691
131+
''')
132+
133+
buf = io.StringIO()
134+
save_values(values, buf)
135+
buf.seek(0)
136+
137+
assert buf.read() == txt
138+
139+
txt = textwrap.dedent('''\
140+
date_thresh=2020-08-08
141+
offset_mean=1.39
142+
offset_stdev=1.07
143+
offset_upper=3.53
144+
offset_lower=-0.75
145+
sugg_src=857.573
146+
sugg_dst=6.8606
147+
''')
148+
149+
buf = io.StringIO()
150+
save_values(values, buf, '{:.2f}', '{:.3f}', '{:.4f}')
151+
buf.seek(0)
152+
153+
assert buf.read() == txt
112154

113155

114156
def test_compute_stuff():

0 commit comments

Comments
 (0)