

def test_sum_with_axis_argument():
    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 (chunk.schema == a.schema)
    assert (agg_expr.dshape == a.x.sum(axis=0).dshape)
    assert chunk_expr.isidentical(chunk.x.sum(axis=0, keepdims=True))
    assert agg_expr.isidentical(agg.sum(axis=0))

dataset/ETHPy150Open blaze/blaze/blaze/expr/tests/test_split.py test_sum_with_axis_argument/original