def test_map(self): def increment(x): return x + 1 add_list = lambda x, y: x + y self.assertListEqual(list(py_builtins.map_(increment, [4, 5, 6])), [5, 6, 7]) self.assertListEqual( list(py_builtins.map_(add_list, [3, 2, 1], [-1, -2, -3])), [2, 0, -2])
def test_map_multiple_datasets(self): add_list = lambda x, y: x + y ds1 = dataset_ops.DatasetV2.from_tensor_slices([-11, -12, 4]) ds2 = dataset_ops.DatasetV2.from_tensor_slices([-21, -22, 5]) ds3 = py_builtins.map_(add_list, ds1, ds2) iterator = dataset_ops.make_one_shot_iterator(ds3) with self.cached_session() as sess: self.assertAllEqual(self.evaluate(iterator.get_next()), -32) self.assertAllEqual(self.evaluate(iterator.get_next()), -34) self.assertAllEqual(self.evaluate(iterator.get_next()), 9)
def test_map_dataset(self): def increment(x): return x + 1 ds1 = dataset_ops.DatasetV2.from_tensor_slices([4, 5, 6]) ds2 = py_builtins.map_(increment, ds1) iterator = dataset_ops.make_one_shot_iterator(ds2) with self.cached_session() as sess: self.assertAllEqual(self.evaluate(iterator.get_next()), 5) self.assertAllEqual(self.evaluate(iterator.get_next()), 6) self.assertAllEqual(self.evaluate(iterator.get_next()), 7)