def METHOD_NAME(setup):
    rs = np.random.RandomState(0)
    raw = rs.randint(5, size=(8, 8, 8))
    arr = tensor(raw, chunk_size=4)
    res = (arr > 3).sum(axis=1).execute().fetch()
    np.testing.assert_array_equal(res, (raw > 3).sum(axis=1))
    res = (arr > 3).sum().execute().fetch()
    np.testing.assert_array_equal(res, (raw > 3).sum())