Test size

Author: hugovk

tests/test_size.py

@@ -0,0 +1,39 @@
+import pytest
+
+import size
+
+
+@pytest.mark.parametrize(
+    ["filename", "expected"],
+    [
+        ("file.tgz", False),
+        ("file.tar.bz2", False),
+        ("file.tar.xz", False),
+        ("file.pdb.zip", False),
+        ("file.amd64.msi", False),
+        ("file.msi", False),
+        ("file.chm", False),
+        ("file.dmg", False),
+        ("file.ext", True),
+    ],
+)
+def test_ignore(filename: str, expected: bool) -> None:
+    assert size.ignore(filename) is expected
+
+
+@pytest.mark.parametrize(
+    ["filename", "expected"],
+    [
+        ("file.tgz", 0),
+        ("file.tar.bz2", 1),
+        ("file.tar.xz", 2),
+        ("file.pdb.zip", 3),
+        ("file.amd64.msi", 4),
+        ("file.msi", 5),
+        ("file.chm", 6),
+        ("file.dmg", 7),
+        ("file.ext", 9999),
+    ],
+)
+def test_key(filename: str, expected: int) -> None:
+    assert size.key(filename) == expected