initial commit

2023-02-18 07:39:08 +01:00
commit e63ab7b77e
63 changed files with 7122 additions and 0 deletions
@@ -0,0 +1,60 @@
+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