

def test_split_reasons_correctly_about_uneven_aggregate_shape():
    x = symbol('chunk', '10 * 10 * int')
    chunk = symbol('chunk', '3 * 3 * int')
    ((chunk, chunk_expr), (agg, agg_expr)) = split(x, x.sum(axis=0), chunk=chunk)
    assert (agg.shape == (4, 10))
    x = symbol('leaf', '1643 * 60 * int')
    chunk = symbol('chunk', '40 * 60 * int')
    ((chunk, chunk_expr), (agg, agg_expr)) = split(chunk_expr, x.sum(), chunk=chunk)
    assert (agg.shape == (42, 1))

dataset/ETHPy150Open blaze/blaze/blaze/expr/tests/test_split.py test_split_reasons_correctly_about_uneven_aggregate_shape/VarMisuse@106/175 `x`->`chunk_expr`