コード例 #1
0
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"]
コード例 #2
0
def test_read_restaurants():
    result = restaurants.read_restaurants(FILENAME)

    assert len(result) == 17
    assert isinstance(result[0], restaurants.Restaurant)