def test_unflatten(): l = [[[10, 20, 30], [40, 50, 60]], [[70, 80, 90], [100, 110, 120]]] labels = enumerate_flat(l) assert unflatten(flatten(l), labels) == l
def test_enumerate_flat(): l = [[[10], [20, 30]], 40] labels = enumerate_flat(l) assert_equal(labels, [[[0], [1, 2]], 3])