diff --git a/app/forms.py b/app/forms.py index 6631a79..c6d735b 100644 --- a/app/forms.py +++ b/app/forms.py @@ -4,5 +4,9 @@ from wtforms import SelectField, SubmitField class DatasetForm(FlaskForm): + """ + Web form to select a dataset + """ + dataset = SelectField(choices=CHOICES) submit = SubmitField("Submit") diff --git a/tests/preprocessing_test.py b/tests/preprocessing_test.py index 74d7886..7080220 100644 --- a/tests/preprocessing_test.py +++ b/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 +from constants import COLUMNS, DATASETS, DATASET_URL, FLICKR_URL def test_dataset_request(): @@ -23,3 +23,11 @@ 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