diff options
Diffstat (limited to 'tests/scanner/test_scanner.py')
| -rw-r--r-- | tests/scanner/test_scanner.py | 23 |
1 files changed, 11 insertions, 12 deletions
diff --git a/tests/scanner/test_scanner.py b/tests/scanner/test_scanner.py index a70c4d8..6fc6650 100644 --- a/tests/scanner/test_scanner.py +++ b/tests/scanner/test_scanner.py @@ -104,18 +104,17 @@ async def test_scanner_dedups_archive_contents(archive, scanner, capsys): archive = await DB.add(archive) dedup_path = archive.path + ".dedup" - with ZipFile(archive.path, "r") as zin: - with ZipFile(dedup_path, "w") as zout: - for info in zin.infolist(): - base, ext = os.path.splitext(info.filename) - - if base == "03": - continue - - if ext == ".png": - zout.writestr(f"0{base}.png", zin.read(info)) - else: - zout.writestr(info.filename, zin.read(info)) + with ZipFile(archive.path, "r") as zin, ZipFile(dedup_path, "w") as zout: + for info in zin.infolist(): + base, ext = os.path.splitext(info.filename) + + if base == "03": + continue + + if ext == ".png": + zout.writestr(f"0{base}.png", zin.read(info)) + else: + zout.writestr(info.filename, zin.read(info)) await scanner.scan() added_archive = await DB.get(Archive, 2, full=True) |