def test_array_field(): field = ArrayField(Var({ 'role_1': a, 'role_2': b, 'role_none': None, }), additional_items=Var({ 'role_3': c, 'role_4': d, 'role_1': e, 'role_none': None, })) assert set(field.iter_all_fields()) == set([a, b, c, d, e]) assert set(field.iter_fields('role_1')) == set([a, e]) assert set(field.iter_fields('role_3')) == set([c]) assert set(field.iter_fields('role_none')) == set([]) field = ArrayField(Var({ 'role_1': (a, b), 'role_2': c }), additional_items=d) assert set(field.iter_all_fields()) == set([a, b, c, d]) field = ArrayField((Var({'role_1': a, 'role_2': b, 'role_none': None}), c)) assert set(field.iter_all_fields()) == set([a, b, c]) assert set(field.iter_fields('role_1')) == set([a, c]) assert set(field.iter_fields('role_none')) == set([c]) field = ArrayField(a, additional_items=b) assert set(field.iter_all_fields()) == set([a, b]) assert set(field.iter_fields('some_role')) == set([a, b]) field = ArrayField() assert set(field.iter_all_fields()) == set([])
def test_array_field(): field = ArrayField(Var({ 'role_1': a, 'role_2': b, }), additional_items=Var({ 'role_3': c, 'role_4': d, })) assert set(field.iter_all_fields()) == set([a, b, c, d]) field = ArrayField(Var({ 'role_1': (a, b), 'role_2': c }), additional_items=d) assert set(field.iter_all_fields()) == set([a, b, c, d]) field = ArrayField(a, additional_items=b) assert set(field.iter_all_fields()) == set([a, b]) field = ArrayField() assert set(field.iter_all_fields()) == set([])