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