def dfjson():
"""return a json representation of the dataframe"""
competitors = []
competitors.append(Comviq("Comviq"))
prices = get_data(competitors)
Large_df = pd.concat(prices, ignore_index=True)
return Response(Large_df.to_json(orient="records"),
mimetype='application/json')
def gettele2():
"""Return Tele2 prices"""
competitors = []
competitors.append(Comviq("Comviq"))
prices = get_data(competitors)
Large_df = pd.concat(prices, ignore_index=True)
return Response(Large_df.to_json(orient="records"),
mimetype='application/json')
def getall():
"""Return all prices"""
competitors = []
competitors.append(Comviq("Comviq"))
competitors.append(Fello("Fello"))
competitors.append(Hallon("Hallon"))
competitors.append(Halebop("Halebop"))
competitors.append(Tele2("Tele2"))
competitors.append(Telia("Telia"))
competitors.append(Tre("Tre"))
result = get_data(competitors)
Large_df = pd.concat(result, ignore_index=True)
return Response(Large_df.to_json(orient="records"),
mimetype='application/json')
"""
comviq.py unit tests
"""
import requests
from comviq import Comviq
comviq = Comviq("Comviq")
df = comviq.get_dataframe()
function_returns_list = comviq.process_dataframe(df)
def test_site_availability():
"""Test site availability"""
site = Comviq.website
req = requests.head(site)
assert req.status_code == 200
def test_getting_dataframe():
"""Test that dataframe is returning something"""
assert len(df) != 0
def test_processing_dataframe():
"""Test that dataframe is correctly processed"""
assert len(function_returns_list) != 0
def test_number_of_rows():
"""Test that dataframe rows is more than 3"""