Test Case 5: index == num_shards.
| 133 | |
| 134 | // Test Case 5: index == num_shards. |
| 135 | TestCase TestCase5() { |
| 136 | return {/*range_data_param*/ {0, 10, 1}, |
| 137 | /*num_shards*/ |
| 138 | CreateTensor<int64>(TensorShape({}), {5}), |
| 139 | /*index*/ |
| 140 | CreateTensor<int64>(TensorShape({}), {4}), |
| 141 | /*require_non_empty*/ true, |
| 142 | /*expected_outputs*/ |
| 143 | {CreateTensor<int64>(TensorShape({}), {4}), |
| 144 | CreateTensor<int64>(TensorShape({}), {9})}, |
| 145 | /*expected_output_dtypes*/ {DT_INT64}, |
| 146 | /*expected_output_shapes*/ {PartialTensorShape({})}, |
| 147 | /*expected_cardinality*/ 2, |
| 148 | /*breakpoints*/ {0, 1, 5}}; |
| 149 | } |
| 150 | |
| 151 | // Test Case 6: similar with test_case_5 but the number of outputs could not be |
| 152 | // divided evenly by num_shards. |
no test coverage detected