Index: tests/test_rpy.py |
=================================================================== |
--- a/tests/test_rpy.py |
+++ b/tests/test_rpy.py |
@@ -130,11 +130,12 @@ |
@pytest.mark.parametrize('line_type', list(_TEST_EXAMPLES.keys())) |
def test_line2dict_format(line_type): |
"""Test that the API result has the appropriate format. |
Checks for both keys and datatypes. |
""" |
- data = line2dict(_TEST_EXAMPLES[line_type]['in']) |
+ mode = 'start' if line_type == 'header' else 'body' |
+ data = line2dict(_TEST_EXAMPLES[line_type]['in'], mode) |
assert data == _TEST_EXAMPLES[line_type]['out'] |