diff --git a/src/nwb_benchmarks/benchmarks/time_download.py b/src/nwb_benchmarks/benchmarks/time_download.py index 54adccc..77cea76 100644 --- a/src/nwb_benchmarks/benchmarks/time_download.py +++ b/src/nwb_benchmarks/benchmarks/time_download.py @@ -12,7 +12,7 @@ from .params_remote_download import hdf5_params, lindi_remote_rfs_params, zarr_params -class BaseDownloadDandiAPIBenchmark(BaseBenchmark): +class BaseDandiAPIDownloadBenchmark(BaseBenchmark): """ Base class for timing the download of remote NWB files using the DANDI API. """ @@ -27,7 +27,7 @@ def teardown(self, params: dict[str, str]): self.tmpdir.cleanup() -class HDF5DownloadDandiAPIBenchmark(BaseDownloadDandiAPIBenchmark): +class HDF5DandiAPIDownloadBenchmark(BaseDandiAPIDownloadBenchmark): """ Time the download of remote HDF5 NWB files using the DANDI API. """ @@ -41,7 +41,7 @@ def time_download_hdf5_dandi_api(self, params: dict[str, str]): download(urls=params["https_url"], output_dir=self.tmpdir.name) -class ZarrDownloadDandiAPIBenchmark(BaseDownloadDandiAPIBenchmark): +class ZarrDandiAPIDownloadBenchmark(BaseDandiAPIDownloadBenchmark): """ Time the download of remote Zarr NWB files using the DANDI API. """ @@ -55,7 +55,7 @@ def time_download_zarr_dandi_api(self, params: dict[str, str]): download(urls=params["https_url"], output_dir=self.tmpdir.name) -class LindiDownloadDandiAPIBenchmark(BaseDownloadDandiAPIBenchmark): +class LindiDandiAPIDownloadBenchmark(BaseDandiAPIDownloadBenchmark): """ Time the download of a remote LINDI JSON file. """