def testAssociativeScanUnstructured1000Reverse(self):
    data = (np.arange(1000) + 32)
    expected = np.cumsum(data[::(- 1)])[::(- 1)]
    result = lax.associative_scan(operator.add, data, reverse=True)
    self.assertAllClose(result, expected, check_dtypes=False)
def testAssociativeScanUnstructured1000Reverse(self):
    data = (np.arange(1000) + 32)
    expected = np.cumsum(data[::(- 1)])[::(- 1)]
    result = lax.associative_scan(operator.add, data, reverse=True)
    self.assertAllClose(result, expected, check_dtypes=False)