(count)
| 32 | components = (np.arange(10),) |
| 33 | |
| 34 | def do_test(count): |
| 35 | dataset = dataset_ops.Dataset.from_tensor_slices(components).skip(count) |
| 36 | self.assertEqual( |
| 37 | [c.shape[1:] for c in components], |
| 38 | [shape for shape in dataset_ops.get_legacy_output_shapes(dataset)]) |
| 39 | start_range = min(count, 10) if count != -1 else 10 |
| 40 | self.assertDatasetProduces( |
| 41 | dataset, |
| 42 | [tuple(components[0][i:i + 1]) for i in range(start_range, 10)]) |
| 43 | |
| 44 | # Skip fewer than input size, we should skip |
| 45 | # the first 4 elements and then read the rest. |
nothing calls this directly
no test coverage detected