def test_add_data(): section = ExperimentSection(make_tree(['session', 'block', 'trial'], {}), ChainMap()) section.add_data(dict(test=True)) yield check_test_data, section for block in section: yield check_test_data, block for trial in block: yield check_test_data, trial