(self)
| 41 | class ReduceTest(test_base.DatasetTestBase, parameterized.TestCase): |
| 42 | |
| 43 | def testSum(self): |
| 44 | for i in range(10): |
| 45 | ds = dataset_ops.Dataset.range(1, i + 1) |
| 46 | result = ds.reduce(np.int64(0), lambda x, y: x + y) |
| 47 | self.assertEqual(((i + 1) * i) // 2, self.evaluate(result)) |
| 48 | |
| 49 | def testSumTuple(self): |
| 50 |