|
10 | 10 | sys.path.insert(0, str(PROJECT_DIR))
|
11 | 11 |
|
12 | 12 | from tstickers.convert import Backend
|
13 |
| -from tstickers.downloader import StickerDownloader |
| 13 | +from tstickers.manager import StickerManager |
14 | 14 |
|
15 | 15 | token = ""
|
16 | 16 | for candidate in [PROJECT_DIR / "env.txt", PROJECT_DIR / "env"]:
|
|
24 | 24 | raise RuntimeError(msg)
|
25 | 25 |
|
26 | 26 |
|
27 |
| -downloader = StickerDownloader(token) |
28 |
| -downloader.cwd = Path(THISDIR) / "data" |
| 27 | +stickerManager = StickerManager(token) |
| 28 | +stickerManager.cwd = Path(THISDIR) / "data" |
29 | 29 |
|
30 | 30 | packs = [{"pack": "DonutTheDog", "len": 31}]
|
31 | 31 |
|
32 | 32 |
|
33 | 33 | def test_getPack() -> None:
|
34 |
| - stickerPack = downloader.getPack(packs[0]["pack"]) |
| 34 | + stickerPack = stickerManager.getPack(packs[0]["pack"]) |
35 | 35 | assert stickerPack is not None
|
36 | 36 | assert len(stickerPack["files"]) == packs[0]["len"]
|
37 | 37 |
|
38 | 38 |
|
39 | 39 | def test_downloadPack() -> None:
|
40 |
| - stickerPack = downloader.getPack(packs[0]["pack"]) |
41 |
| - assert stickerPack is not None |
42 |
| - downloader.downloadPack(stickerPack) |
43 |
| - assert len(list(Path(f"{downloader.cwd}/donutthedog/tgs").iterdir())) == packs[0]["len"] |
| 40 | + stickerManager.downloadPack(packs[0]["pack"]) |
| 41 | + assert len(list(Path(f"{stickerManager.cwd}/donutthedog/tgs").iterdir())) == packs[0]["len"] |
44 | 42 |
|
45 | 43 |
|
46 | 44 | def test_convertPack() -> None:
|
47 |
| - stickerPack = downloader.getPack(packs[0]["pack"]) |
48 |
| - assert stickerPack is not None |
49 |
| - downloader.downloadPack(stickerPack) |
50 |
| - downloader.convertPack(packs[0]["pack"], scale=0.05, noCache=True, backend=Backend.PYRLOTTIE) |
51 |
| - assert len(list(Path(f"{downloader.cwd}/donutthedog/webp").iterdir())) == packs[0]["len"] |
| 45 | + stickerManager.downloadPack(packs[0]["pack"]) |
| 46 | + stickerManager.convertPack(packs[0]["pack"], scale=0.05, noCache=True, backend=Backend.RLOTTIE_PYTHON) |
| 47 | + assert len(list(Path(f"{stickerManager.cwd}/donutthedog/webp").iterdir())) == packs[0]["len"] |
52 | 48 |
|
53 | 49 |
|
54 | 50 | # def test_convertPack_slow() -> None:
|
55 |
| -# stickerPack = downloader.getPack(packs[0]["pack"]) |
56 |
| -# assert stickerPack is not None |
57 |
| -# downloader.downloadPack(stickerPack) |
58 |
| -# downloader.convertPack(packs[0]["pack"], scale=1, noCache=True, backend=Backend.PYRLOTTIE) |
59 |
| -# assert len(list(Path(f"{downloader.cwd}/donutthedog/webp").iterdir())) == packs[0]["len"] |
| 51 | +# stickerManager.downloadPack(packs[0]["pack"]) |
| 52 | +# stickerManager.convertPack(packs[0]["pack"], scale=1, noCache=True, backend=Backend.PYRLOTTIE) |
| 53 | +# assert len(list(Path(f"{stickerManager.cwd}/donutthedog/webp").iterdir())) == packs[0]["len"] |
0 commit comments