Function test_map_grouping_mixed

tests/unit/library/test_grouping.py:157–167

(mixed_grouping_size)

def test_map_grouping_mixed(mixed_grouping_size):
    grouping, size = mixed_grouping_size

    def fn(x):
        return x * 2 + 5

    result = map_grouping(fn, grouping)
    expected = make_grouping_by_index(
        grouping, list(map(fn, flatten_grouping(grouping)))
    )
    assert expected == result


# Test make_grouping_by_key