diff --git a/app/main.py b/app/main.py index 4850836f..7ea4189e 100644 --- a/app/main.py +++ b/app/main.py @@ -1,6 +1,6 @@ class Car: - def __init__(self, comfort_class: str, clean_mark: str, brand: str)\ - -> None: + def __init__(self, comfort_class: int, clean_mark: int, brand: str) -> \ + None: self.comfort_class = comfort_class self.clean_mark = clean_mark self.brand = brand @@ -14,16 +14,16 @@ def __init__(self, distance_from_city_center: int, clean_power: int, self.average_rating = average_rating self.count_of_ratings = count_of_ratings - def calculate_washing_price(self, car: str) -> None: + def calculate_washing_price(self, car: Car) -> None: return round((car.comfort_class * (self.clean_power - car.clean_mark) * self.average_rating) / self.distance_from_city_center, 1) - def wash_single_car(self, car: str) -> None: + def wash_single_car(self, car: Car) -> None: if car.clean_mark < self.clean_power: car.clean_mark = self.clean_power - def serve_cars(self, cars: str) -> None: + def serve_cars(self, cars: Car) -> None: income = 0 for car in cars: if car.clean_mark < self.clean_power: