Маршрут — список городов, в том числе повторяющихся. Цену билета в город возвращает fetch_price(city), но каждый вызов — медленный запрос к серверу. Функция trip_cost(route, fetch_price) считает полную стоимость маршрута и должна запрашивать цену каждого города не больше одного раза, храня ответы в кэше. В функции две ошибки — найдите и исправьте их.
prices = {"Казань": 3000, "Сочи": 5000}
trip_cost(["Казань", "Сочи", "Казань"], prices.__getitem__) # 11000, два запросаЗаготовка задачи обновилась с тех пор, как вы сохранили свой код.
Сбросить редактор к заготовке? Ваш код будет удалён.
from collections.abc import Callable
def trip_cost(route: list[str], fetch_price: Callable[[str], int]) -> int:
"""Сумма цен по маршруту; цена каждого города запрашивается один раз."""
total = 0
for city in route:
cache: dict[str, int] = {}
if city not in cache:
cache[city] = fetch_price(city)
total += fetch_price(city)
return totalВывод обрезан: показаны первые строки.
PRICES = {"Казань": 3000, "Сочи": 5000, "Омск": 4200, "Тверь": 900}
def _server():
"""Возвращает fetch_price со счётчиком запросов по городам."""
calls: dict[str, int] = {}
def fetch_price(city: str) -> int:
calls[city] = calls.get(city, 0) + 1
return PRICES[city]
return fetch_price, calls
def test_example_total():
"""Пример: Казань, Сочи, Казань → 11000"""
fetch, _ = _server()
got = trip_cost(["Казань", "Сочи", "Казань"], fetch)
assert got == 11000, f"ожидалось 11000, получено {got!r}"
def test_example_calls():
"""Пример: Казань запрошена один раз, хотя встречается дважды"""
fetch, calls = _server()
trip_cost(["Казань", "Сочи", "Казань"], fetch)
assert calls == {"Казань": 1, "Сочи": 1}, f"запросы к серверу: {calls}"
def test_empty():
"""Пустой маршрут → 0, без запросов"""
fetch, calls = _server()
got = trip_cost([], fetch)
assert got == 0 and not calls, f"итог {got!r}, запросы {calls}"
def test_single_city_many_times():
"""Один город пять раз подряд — один запрос"""
fetch, calls = _server()
got = trip_cost(["Тверь"] * 5, fetch)
assert got == 4500, f"ожидалось 4500, получено {got!r}"
assert calls == {"Тверь": 1}, f"запросы к серверу: {calls}"
def test_all_different():
"""Все города разные — по одному запросу на город"""
fetch, calls = _server()
got = trip_cost(list(PRICES), fetch)
assert got == sum(PRICES.values()), f"ожидалось {sum(PRICES.values())}, получено {got!r}"
assert all(n == 1 for n in calls.values()) and len(calls) == 4, f"запросы к серверу: {calls}"
def test_long_route():
"""Длинный маршрут с повторами"""
route = ["Омск", "Сочи", "Омск", "Тверь", "Сочи", "Омск", "Казань", "Тверь"]
fetch, calls = _server()
got = trip_cost(route, fetch)
expected = sum(PRICES[c] for c in route)
assert got == expected, f"ожидалось {expected}, получено {got!r}"
assert sum(calls.values()) == 4, f"сделано {sum(calls.values())} запросов, нужно 4: {calls}"Можно решать и в своём редакторе: «Скопировать с тестами» и запустите python3 trip_cost_cache.py — тесты те же, результат покажет ✓ / ✗ и итог.
Тесты считают вызовы fetch_price. Для каждого города он должен случиться ровно один раз.
Сколько живёт словарь cache? Переживает ли он переход к следующему городу?
Цена записана в кэш — откуда её берут при подсчёте суммы?
from collections.abc import Callable
def trip_cost(route: list[str], fetch_price: Callable[[str], int]) -> int:
"""Сумма цен по маршруту; цена каждого города запрашивается один раз."""
total = 0
cache: dict[str, int] = {}
for city in route:
if city not in cache:
cache[city] = fetch_price(city)
total += cache[city]
return total