Add Flickr avalaibility test
This commit is contained in:
parent
bb610f3935
commit
44785255b0
@ -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")
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user