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):
|
class DatasetForm(FlaskForm):
|
||||||
|
"""
|
||||||
|
Web form to select a dataset
|
||||||
|
"""
|
||||||
|
|
||||||
dataset = SelectField(choices=CHOICES)
|
dataset = SelectField(choices=CHOICES)
|
||||||
submit = SubmitField("Submit")
|
submit = SubmitField("Submit")
|
||||||
|
@ -3,7 +3,7 @@ from requests import get
|
|||||||
|
|
||||||
from app.preprocessing import create_dataframe
|
from app.preprocessing import create_dataframe
|
||||||
from app.data_request import request_dataset
|
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():
|
def test_dataset_request():
|
||||||
@ -23,3 +23,11 @@ def test_dataframe_creation():
|
|||||||
df = create_dataframe(dataset)
|
df = create_dataframe(dataset)
|
||||||
assert isinstance(df, DataFrame)
|
assert isinstance(df, DataFrame)
|
||||||
assert all(df.columns == COLUMNS[dataset])
|
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