def test_walk(): session = ExperimentSection(make_tree(['session', 'block', 'trial'], {}), ChainMap()) all_sections = [session] for block in session: all_sections.append(block) all_sections.extend(block) assert all_sections == list(session.walk())