| 123456789101112131415 |
- from anonymizer.detection import download_weights
- class TestDownloadWeights:
- @staticmethod
- def test_it_downloads_weights(tmp_path):
- weights_directory = tmp_path / 'weights'
- assert len(list(weights_directory.glob('**/*.pb'))) == 0
- download_weights(download_directory=weights_directory, version='1.0.0')
- assert len(list(weights_directory.glob('**/*.pb'))) == 2
- assert (weights_directory / 'weights_face_v1.0.0.pb').is_file()
- assert (weights_directory / 'weights_plate_v1.0.0.pb').is_file()
- assert not (weights_directory / 'nonexistent_path.pb').is_file()
|
