Function TestCase7

tensorflow/core/kernels/data/shard_dataset_op_test.cc:171–184

Test Case 7: num_shard is larger than the cardinality of input dataset; require_non_empty = false.

// Test Case 7: num_shard is larger than the cardinality of input dataset;
// require_non_empty = false.
TestCase TestCase7() {
  return {/*range_data_param*/ {0, 10, 1},
          /*num_shards*/
          CreateTensor<int64>(TensorShape({}), {20}),
          /*index*/
          CreateTensor<int64>(TensorShape({}), {5}),
          /*require_non_empty*/ false,
          /*expected_outputs*/
          {CreateTensor<int64>(TensorShape({}), {5})},
          /*expected_output_dtypes*/ {DT_INT64},
          /*expected_output_shapes*/ {PartialTensorShape({})},
          /*expected_cardinality*/ 1,
          /*breakpoints*/ {0, 5}};
}

// Test Case 8: similar with test_case_7 but require_non_empty = true.
TestCase NoElemForEachShardTestCase() {