

def test_split_reasons_correctly_about_aggregate_shape():
    chunk = symbol('chunk', '100 * 100 * {x: float32, y: float32}')
    ((chunk, chunk_expr), (agg, agg_expr)) = split(a, a.x.sum(), chunk=agg)
    assert (agg.shape == (10, 20))
    chunk = symbol('chunk', '100 * 100 * {x: float32, y: float32}')
    ((chunk, chunk_expr), (agg, agg_expr)) = split(a, a.x.sum(axis=0), chunk=chunk)
    assert (agg.shape == (10, 2000))

dataset/ETHPy150Open blaze/blaze/blaze/expr/tests/test_split.py test_split_reasons_correctly_about_aggregate_shape/VarMisuse@46/83 `chunk`->`agg`