Add Flickr avalaibility test

This commit is contained in:
coolneng 2020-06-15 02:14:04 +02:00
parent bb610f3935
commit 44785255b0
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
2 changed files with 13 additions and 1 deletions

View File

@ -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")

View File

@ -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