|
a |
|
b/tests/spec/test_conftest.py |
|
|
1 |
from ehrql.query_model.nodes import Column, TableSchema |
|
|
2 |
|
|
|
3 |
from .conftest import parse_row, parse_table |
|
|
4 |
|
|
|
5 |
|
|
|
6 |
def test_parse_table(): |
|
|
7 |
assert parse_table( |
|
|
8 |
TableSchema(i1=Column(int), i2=Column(int)), |
|
|
9 |
""" |
|
|
10 |
| i1 | i2 |
|
|
11 |
--+-----+----- |
|
|
12 |
1 | 101 | 111 |
|
|
13 |
2 | 201 | |
|
|
14 |
""", |
|
|
15 |
) == [ |
|
|
16 |
{"patient_id": 1, "i1": 101, "i2": 111}, |
|
|
17 |
{"patient_id": 2, "i1": 201, "i2": None}, |
|
|
18 |
] |
|
|
19 |
|
|
|
20 |
|
|
|
21 |
def test_parse_row(): |
|
|
22 |
assert parse_row( |
|
|
23 |
{"patient_id": int, "i1": int, "i2": int}, |
|
|
24 |
["patient_id", "i1", "i2"], |
|
|
25 |
"1 | 101 | 111", |
|
|
26 |
) == {"patient_id": 1, "i1": 101, "i2": 111} |