def test_next_normal(self): iterator = iter([1, 2, 3]) self.assertEqual(py_builtins.next_(iterator), 1) self.assertEqual(py_builtins.next_(iterator), 2) self.assertEqual(py_builtins.next_(iterator), 3) with self.assertRaises(StopIteration): py_builtins.next_(iterator) self.assertEqual(py_builtins.next_(iterator, 4), 4)
def test_fn(go_out_of_range, with_default): iterator = iter(dataset_ops.Dataset.range(3)) retval = ( py_builtins.next_(iterator), py_builtins.next_(iterator), py_builtins.next_(iterator), ) if go_out_of_range: if with_default: retval += ( py_builtins.next_(iterator, constant_op.constant(-3, dtype=dtypes.int64)), py_builtins.next_(iterator, constant_op.constant(-4, dtype=dtypes.int64)), ) else: py_builtins.next_(iterator) return retval
def test_fn(default_val): ds = dataset_ops.Dataset.range(1) ds = ds.map(lambda i: {'a': i + 1, 'b': i + 10}) iterator = iter(ds) py_builtins.next_(iterator) py_builtins.next_(iterator, default_val)
def test_fn(): iterator = iter(dataset_ops.Dataset.range(1)) py_builtins.next_(iterator) py_builtins.next_(iterator, constant_op.constant(-3))