Method testZipDataset

tensorflow/python/data/kernel_tests/zip_test.py:33–68

(self)

class ZipTest(test_base.DatasetTestBase):

  def testZipDataset(self):

    def dataset_fn(components):
      datasets = tuple([
          dataset_ops.Dataset.from_tensor_slices(component)
          for component in components
      ])
      return dataset_ops.Dataset.zip(datasets)

    equal_length_components = [
        np.tile(np.array([[1], [2], [3], [4]]), 20),
        np.tile(np.array([[12], [13], [14], [15]]), 22),
        np.array([37.0, 38.0, 39.0, 40.0])
    ]

    get_next = self.getNext(dataset_fn(equal_length_components))
    for i in range(4):
      results = self.evaluate(get_next())
      for component, result_component in zip(equal_length_components, results):
        self.assertAllEqual(component[i], result_component)
    with self.assertRaises(errors.OutOfRangeError):
      self.evaluate(get_next())
    with self.assertRaises(errors.OutOfRangeError):
      self.evaluate(get_next())

    variable_length_components = [[1, 2, 3, 4], [1, 2, 3, 4, 5], [1.0, 2.0]]
    get_next = self.getNext(dataset_fn(variable_length_components))
    for i in range(2):
      results = self.evaluate(get_next())
      for component, result_component in zip(variable_length_components,
                                             results):
        self.assertAllEqual(component[i], result_component)
    with self.assertRaises(errors.OutOfRangeError):
      self.evaluate(get_next())
    with self.assertRaises(errors.OutOfRangeError):
      self.evaluate(get_next())

  def testNestedZipDataset(self):