Allow capitalized extensions, add basic tests

README.md

@@ -98,6 +98,10 @@ tar -xvf $(pwd)/data/${FILENAME} -C $(pwd)/data/
 
 A third dataset from the Lincoln Park Zoo's Urban Wildlife Institute contains unlabeled three-image bursts from 2018. It takes up 5.7GB uncompressed. To get this data, follow the same steps as for the 2012-2014 dataset, but replace `FILENAME=lpz_2016_2017_processed.tar.gz` with `FILENAME=lpz_2018.tar.gz`.
 
+## Running Tests
+
+To test the app, run `pip install -r requirements-dev.txt` and then `pytest`. The tests assume that the app is running locally on port `8000` according to the instructions above.``
+
 ## Example Images
 
 ![buck](./gallery/buck.jpeg)

autofocus/predict/app/utils.py

@@ -1,3 +1,4 @@
+from pathlib import Path
 from zipfile import ZipFile
 
 
@@ -11,7 +12,7 @@ def allowed_file(filename, allowed_extensions):
         bool: whether the filename is in allowed extensions
 
     """
-    return "." in filename and filename.rsplit(".", 1)[1] in allowed_extensions
+    return Path(filename).suffix.lower().replace(".", "") in allowed_extensions
 
 
 def list_zip_files(path):

requirements-dev.txt

@@ -0,0 +1,6 @@
+black
+flake8
+flake8-docstrings
+flake8-import-order
+pydocstyle<4.0.0
+pytest