diff --git a/tests/test_extended.py b/tests/test_extended.py deleted file mode 100644 index 663e4ef..0000000 --- a/tests/test_extended.py +++ /dev/null @@ -1,24 +0,0 @@ -import pytest - -from PTT.handlers import add_defaults -from PTT.parse import Parser - - -@pytest.fixture -def parser(): - p = Parser() - add_defaults(p) - return p - - -@pytest.mark.parametrize("release_name, expected_extended", [ - ("Have I Got News For You S53E02 EXTENDED 720p HDTV x264-QPEL", True), - ("Better.Call.Saul.S03E04.CONVERT.720p.WEB.h264-TBS", False), -]) -def test_extended_detection(parser, release_name, expected_extended): - result = parser.parse(release_name) - assert isinstance(result, dict), f"Parser did not return a dict for {release_name}" - if expected_extended: - assert result.get("extended") == expected_extended, f"Failed to detect 'extended' correctly for {release_name}" - else: - assert "extended" not in result, f"Incorrectly detected 'extended' for {release_name}" diff --git a/tests/test_main.py b/tests/test_main.py index 5fa4cae..408f573 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -416,6 +416,22 @@ def parser(): "channels": "5.1", "codec": "avc", "group": "LEGi0N" + }), + ("The.Lord.of.the.Rings.The.Fellowship.of.the.Ring.2001.EXTENDED.2160p.UHD.BluRay.x265.10bit.HDR.TrueHD.7.1.Atmos-BOREDOR", { + "title": "The Lord of the Rings The Fellowship of the Ring", + "year": 2001, + "resolution": "4k", + "edition": "Extended Edition", + "languages": [], + "seasons": [], + "episodes": [], + "quality": "BluRay", + "codec": "x265", + "bit_depth": "10bit", + "audio": "7.1 Atmos", + "channels": "7.1", + "hdr": ["HDR"], + "group": "BOREDOR" }) ]) def test_random_releases_parse(parser, release_name, expected):