|
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