message: Add test case for "broadcast" padding_values to the same structure as dataset
diff: @@ -306,6 +306,22 @@ class PaddedBatchTest(test_base.DatasetTestBase, parameterized.TestCase):
  self.assertDatasetProduces(
  ds, expected_output=[[0.0, 1.0, 2.0, 3.0, 4.0]])
  + @combinations.generate(test_base.default_test_combinations())
  + def testDefaultPaddedValueShapes(self):
  +
  + def fill(x):
  + return array_ops.fill([x], x)
  +
  + dataset = dataset_ops.Dataset.zip((
  + dataset_ops.Dataset.from_tensor_slices([1, 2, 3, 4]).map(fill),
  + dataset_ops.Dataset.from_tensor_slices([1, 2, 3, 4]).map(fill)))
  + dataset = dataset.padded_batch(batch_size=2, padding_values=-1)
  + self.assertDatasetProduces(
  + dataset,
  + expected_output=[
  + ([[1, -1], [2, 2]], [[1, -1], [2, 2]]),
  + ([[3, 3, 3, -1], [4, 4, 4, 4]], [[3, 3, 3, -1], [4, 4, 4, 4]])])
  +
  if __name__ == '__main__':
  test.main()
  