Merge pull request #113 from hugovk/add-tests-size

Author: hugovk

size.py

@@ -14,11 +14,11 @@
     ('tgz', 'tar.bz2', 'tar.xz', 'pdb.zip', 'amd64.msi', 'msi', 'chm', 'dmg'))}
 
 
-def ignore(filename):
+def ignore(filename: str) -> bool:
     return not any(filename.endswith(DOT + ext) for ext in sort_order)
 
 
-def key(filename):
+def key(filename: str) -> int:
     parts = filename.split(DOT)
     # Try 2 parts first.
     ext = DOT.join(parts[-2:])
@@ -28,11 +28,16 @@ def key(filename):
     return sort_order.get(ext, 9999)
 
 
-for filename in sorted(sys.argv[1:], key=key):
-    if ignore(filename):
-        continue
-    md5 = hashlib.md5()
-    with open(filename, 'rb') as fp:
-        md5.update(fp.read())
-    size = os.stat(filename).st_size
-    print(f'  {md5.hexdigest()}  {size:8}  {filename}')
+def main() -> None:
+    for filename in sorted(sys.argv[1:], key=key):
+        if ignore(filename):
+            continue
+        md5 = hashlib.md5()
+        with open(filename, 'rb') as fp:
+            md5.update(fp.read())
+        size = os.stat(filename).st_size
+        print(f'  {md5.hexdigest()}  {size:8}  {filename}')
+
+
+if __name__ == '__main__':
+    main()

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