bites 116

Author: [email protected]

.gitignore

@@ -65,3 +65,5 @@
 /146/README.md
 /140/README.md
 /140-pandas/README.md
+/118/README.md
+/116/README.md

116/files.py

@@ -0,0 +1,13 @@
+import glob
+import os
+
+ONE_KB = 1024
+
+
+def get_files(dirname, size_in_kb):
+    """Return files in dirname that are >= size_in_kb"""
+    file_list = os.listdir(dirname)
+    return sorted([file for file in file_list if os.path.getsize(file) >= size_in_kb])
+
+
+print(get_files(".", ONE_KB))

116/test_files.py

@@ -0,0 +1,26 @@
+import os
+from tempfile import TemporaryDirectory
+
+import pytest
+
+from files import get_files
+
+TMP = '/tmp'
+
+
+@pytest.mark.parametrize("byte_sizes, size_in_kb, expected", [
+    ([800, 1000, 1200], 1, ['1200']),
+    ([1024, 1025], 1, ['1024', '1025']),
+    ([1024, 1025], 1.026, []),
+    ([1000, 1300, 1777, 900], 1.25, ['1300', '1777']),
+    ([1024, 2047, 2048, 2500], 2, ['2048', '2500']),
+])
+def test_get_files(byte_sizes, size_in_kb, expected):
+    with TemporaryDirectory(dir=TMP) as dirname:
+        for size in byte_sizes:
+            with open(os.path.join(dirname, str(size)), 'wb') as f:
+                f.write(os.urandom(size))
+
+        actual = [os.path.basename(fi) for fi in
+                  get_files(dirname, size_in_kb)]
+        assert sorted(actual) == sorted(expected)