| 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'] |