def testLowestRated(self):
fp.init_db()
yelp_data = fp.getYelpData("breakfast", "Ann Arbor")
fp.loadYelpData(yelp_data)
results = fp.getLowestRated()
self.assertEqual(len(results), 5)
for res in results:
self.assertTrue(res[1] <= 3.0)
def testLowestPrice(self):
fp.init_db()
yelp_data = fp.getYelpData("breakfast", "Ann Arbor")
fp.loadYelpData(yelp_data)
results = fp.getLowestPrice()
self.assertEqual(len(results), 5)
for res in results:
self.assertTrue(len(res[4]) <= 2)
self.assertTrue(len(res[4]) != 0)
def testReviews(self):
fp.init_db()
cur.execute("SELECT * FROM reviews")
inital_results = cur.fetchall()
self.assertFalse(inital_results)
yelp_data = fp.getYelpData("breakfast", "Ann Arbor")
fp.loadYelpData(yelp_data)
cur.execute("SELECT * FROM reviews")
results = cur.fetchall()
self.assertTrue(type(results) == list)
self.assertTrue(len(results) >= 100)
self.assertTrue(len(results[0]) == 5)
def testYelpRequest(self):
yelp_data = fp.getYelpData("breakfast", "Ann Arbor")
self.assertTrue(type(yelp_data) == list)
self.assertTrue(len(yelp_data) > 5)
self.assertTrue(type(yelp_data[0]) == fp.Yelp)
self.assertTrue("Ann Arbor" in yelp_data[0].address)