mirror of
https://github.com/nathom/streamrip.git
synced 2024-09-19 19:28:46 -04:00
20 lines
531 B
Python
20 lines
531 B
Python
import os
|
|
import time
|
|
from pprint import pprint
|
|
|
|
from streamrip.downloadtools import DownloadPool
|
|
|
|
|
|
def test_downloadpool(tmpdir):
|
|
start = time.perf_counter()
|
|
with DownloadPool(
|
|
(f"https://pokeapi.co/api/v2/pokemon/{number}" for number in range(1, 151)),
|
|
tempdir=tmpdir,
|
|
) as pool:
|
|
pool.download()
|
|
assert len(os.listdir(tmpdir)) == 151
|
|
|
|
# the tempfiles should be removed at this point
|
|
assert len(os.listdir(tmpdir)) == 0
|
|
|
|
print(f"Finished in {time.perf_counter() - start}s")
|