--- a +++ b/tests/spec/test_conftest.py @@ -0,0 +1,26 @@ +from ehrql.query_model.nodes import Column, TableSchema + +from .conftest import parse_row, parse_table + + +def test_parse_table(): + assert parse_table( + TableSchema(i1=Column(int), i2=Column(int)), + """ + | i1 | i2 + --+-----+----- + 1 | 101 | 111 + 2 | 201 | + """, + ) == [ + {"patient_id": 1, "i1": 101, "i2": 111}, + {"patient_id": 2, "i1": 201, "i2": None}, + ] + + +def test_parse_row(): + assert parse_row( + {"patient_id": int, "i1": int, "i2": int}, + ["patient_id", "i1", "i2"], + "1 | 101 | 111", + ) == {"patient_id": 1, "i1": 101, "i2": 111}