def test_sparse_scan(self): axis = 1 a = expr.sparse_diagonal(ARRAY_SIZE, dtype=np.float32, tile_hint=tile_hint) c = expr.scan(a, reduce_fn=lambda x, **kw:x.sum(axis=kw['axis']), scan_fn=lambda x, **kw: x.cumsum(axis=kw['axis']), accum_fn=None, axis=axis) print c.glom()
def test_sum_scan(self): axis = None a = expr.ones(ARRAY_SIZE, dtype=np.float32, tile_hint=tile_hint) c = expr.scan(a, reduce_fn=np.sum, scan_fn=np.cumsum, accum_fn=None, axis=axis) print c.glom()