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)