def test_recordmapmany():
table = (('id', 'sex', 'age', 'height', 'weight'),
(1, 'male', 16, 1.45, 62.0), (2, 'female', 19, 1.34, 55.4),
(3, '-', 17, 1.78, 74.4), (4, 'male', 21, 1.33))
def rowgenerator(rec):
transmf = {'male': 'M', 'female': 'F'}
yield [
rec['id'], 'gender',
transmf[rec['sex']] if rec['sex'] in transmf else rec['sex']
]
yield [rec['id'], 'age_months', rec['age'] * 12]
yield [rec['id'], 'bmi', rec['weight'] / rec['height']**2]
actual = rowmapmany(table,
rowgenerator,
header=['subject_id', 'variable', 'value'])
expect = (('subject_id', 'variable', 'value'), (1, 'gender', 'M'),
(1, 'age_months', 16 * 12), (1, 'bmi', 62.0 / 1.45**2),
(2, 'gender', 'F'), (2, 'age_months', 19 * 12),
(2, 'bmi', 55.4 / 1.34**2), (3, 'gender', '-'),
(3, 'age_months', 17 * 12), (3, 'bmi', 74.4 / 1.78**2),
(4, 'gender', 'M'), (4, 'age_months', 21 * 12))
ieq(expect, actual)
ieq(expect, actual) # can iteratate twice?
def test_recordmapmany():
table = (('id', 'sex', 'age', 'height', 'weight'),
(1, 'male', 16, 1.45, 62.0),
(2, 'female', 19, 1.34, 55.4),
(3, '-', 17, 1.78, 74.4),
(4, 'male', 21, 1.33))
def rowgenerator(rec):
transmf = {'male': 'M', 'female': 'F'}
yield [rec['id'], 'gender',
transmf[rec['sex']] if rec['sex'] in transmf else rec['sex']]
yield [rec['id'], 'age_months', rec['age'] * 12]
yield [rec['id'], 'bmi', rec['weight'] / rec['height'] ** 2]
actual = rowmapmany(table, rowgenerator, header=['subject_id', 'variable',
'value'])
expect = (('subject_id', 'variable', 'value'),
(1, 'gender', 'M'),
(1, 'age_months', 16 * 12),
(1, 'bmi', 62.0 / 1.45 ** 2),
(2, 'gender', 'F'),
(2, 'age_months', 19 * 12),
(2, 'bmi', 55.4 / 1.34 ** 2),
(3, 'gender', '-'),
(3, 'age_months', 17 * 12),
(3, 'bmi', 74.4 / 1.78 ** 2),
(4, 'gender', 'M'),
(4, 'age_months', 21 * 12))
ieq(expect, actual)
ieq(expect, actual) # can iteratate twice?
