| 203 | |
| 204 | |
| 205 | def test_build_overwrite(build, default_data_dir: epath.Path): |
| 206 | data_dir = default_data_dir / 'mnist/3.0.1' |
| 207 | data_dir.mkdir(parents=True) |
| 208 | metadata_path = tfds.core.tfds_path( |
| 209 | 'testing/test_data/dataset_info/mnist/3.0.1' |
| 210 | ) |
| 211 | |
| 212 | for f in metadata_path.iterdir(): # Copy metadata files. |
| 213 | data_dir.joinpath(f.name).write_text(f.read_text()) |
| 214 | |
| 215 | # By default, will skip generation if the data already exists |
| 216 | assert build('mnist') == ['mnist'] # Called, but no-op |
| 217 | assert data_dir.exists() |
| 218 | |
| 219 | assert build('mnist --overwrite') == ['mnist'] |
| 220 | assert not data_dir.exists() # Previous data-dir has been removed |
| 221 | |
| 222 | |
| 223 | def test_max_examples_per_split_0(build, default_data_dir: epath.Path): |