def test_group_restaurants(): rests = restaurants.read_restaurants(FILENAME) (ratings, prices, cuisines) = restaurants.group_restaurants(rests) assert len(ratings) == 17 assert len(prices) == 4 assert len(cuisines) == 13 assert ratings["Queen St. Cafe"] == 82 assert "Queen St. Cafe" in prices["$"] assert "Queen St. Cafe" in cuisines["Malaysian"] assert "Queen St. Cafe" in cuisines["Thai"]
def test_read_restaurants(): result = restaurants.read_restaurants(FILENAME) assert len(result) == 17 assert isinstance(result[0], restaurants.Restaurant)