diff --git a/tests/test_build_docs_version.py b/tests/test_build_docs_version.py index 293388c..5908f4d 100644 --- a/tests/test_build_docs_version.py +++ b/tests/test_build_docs_version.py @@ -1,7 +1,29 @@ from build_docs import Version -def test_filter() -> None: +def test_filter_default() -> None: + # Arrange + versions = [ + Version("3.14", status="feature"), + Version("3.13", status="bugfix"), + Version("3.12", status="bugfix"), + Version("3.11", status="security"), + Version("3.10", status="security"), + Version("3.9", status="security"), + ] + + # Act + filtered = Version.filter(versions) + + # Assert + assert filtered == [ + Version("3.14", status="feature"), + Version("3.13", status="bugfix"), + Version("3.12", status="bugfix"), + ] + + +def test_filter_one() -> None: # Arrange versions = [ Version("3.14", status="feature"),