app/forms.py

@@ -4,9 +4,5 @@ from wtforms import SelectField, SubmitField
 
 
 class DatasetForm(FlaskForm):
-    """
-    Web form to select a dataset
-    """
-
     dataset = SelectField(choices=CHOICES)
     submit = SubmitField("Submit")

shell.nix

@@ -10,8 +10,17 @@ pkgs.mkShell {
     flask
     flask-bootstrap
     flask_wtf
+    matplotlib
     folium
     pytest
     beautifulsoup4
+    # Development tools
+    black
+    isort
+    pyflakes
+    python-language-server
+    pyls-black
+    pyls-isort
+    pyls-mypy
   ];
 }

tests/preprocessing_test.py

@@ -3,7 +3,7 @@ from requests import get
 
 from app.preprocessing import create_dataframe
 from app.data_request import request_dataset
-from constants import COLUMNS, DATASETS, DATASET_URL, FLICKR_URL
+from constants import COLUMNS, DATASETS, DATASET_URL
 
 
 def test_dataset_request():
@@ -23,11 +23,3 @@ def test_dataframe_creation():
         df = create_dataframe(dataset)
         assert isinstance(df, DataFrame)
         assert all(df.columns == COLUMNS[dataset])
-
-
-def test_flickr_request():
-    """
-    Checks that Flickr search is avalaible
-    """
-    response = get(FLICKR_URL.format("paris coronavirus"))
-    assert response.status_code == 200