61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
from cmath import exp
|
|
from torrentify import generate_title, parse_filename, ParsedFilename
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
["filename", "expected"],
|
|
[
|
|
[
|
|
"To Ki -- @Studio -- 2022-10-04",
|
|
{
|
|
"actors": ["To Ki"],
|
|
"date": "2022-10-04",
|
|
"studio": "Studio",
|
|
"tags": ["to.ki", "studio.com", "2022", "2022.10"],
|
|
},
|
|
],
|
|
[
|
|
"To Ki, Ki To -- @Studio -- Title -- 2022-10-04",
|
|
{
|
|
"actors": ["To Ki", "Ki To"],
|
|
"date": "2022-10-04",
|
|
"studio": "Studio",
|
|
"title": "Title",
|
|
"tags": ["to.ki", "ki.to", "studio.com", "2022", "2022.10"],
|
|
},
|
|
],
|
|
],
|
|
)
|
|
def test_parse_filename(filename: str, expected: dict):
|
|
parsed = parse_filename(filename)
|
|
print(filename, parsed)
|
|
assert parsed == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
["parsed", "expected"],
|
|
[
|
|
[
|
|
{
|
|
"actors": ["To Ki"],
|
|
"date": "2022-10-04",
|
|
"studio": "Studio",
|
|
},
|
|
'To Ki -- @Studio -- 2022-10-04',
|
|
],
|
|
[
|
|
{
|
|
"actors": ["To Ki", "Ki To"],
|
|
"date": "2022-10-04",
|
|
"title": "Title",
|
|
"studio": "Studio",
|
|
},
|
|
'To Ki, Ki To -- @Studio -- Title -- 2022-10-04',
|
|
],
|
|
],
|
|
)
|
|
def test_generate_title(parsed: ParsedFilename, expected: str):
|
|
generated = generate_title(parsed)
|
|
assert generated == expected
|