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()
