Написать pytest для класса: class Vehicle: def __init__(self, name, mileage): self.name = name self.mileage = mileage def get_vehicle_type(self, num_wheels): if num_wheels == 2: return f'Это мотоцикл марки {self.name}.' elif num_wheels == 3: return f'Это трицикл марки {self.name}.' elif num_wheels == 4: return f'Это автомобиль марки {self.name}.' else: return 'Я не знаю таких ТС.' def get_vehicle_advice(self): if self.mileage < 50000: return f"Неплохо {self.name} можно брать." elif 50001 <= self.mileage <= 100000: return f"{self.name} надо внимательно проверить." elif 100001 <= self.mileage <= 150000: return f"{self.name} надо провести полную диагностику." else: return f"{self.name} лучше не покупать."

import pytest from vehicle import Vehicle

def test_get_vehicle_type(): vehicle1 = Vehicle("Harley Davidson", 2) assert vehicle1.get_vehicle_type(2) == "Это мотоцикл марки Harley Davidson."

vehicle2 = Vehicle("Can-Am", 3)
assert vehicle2.get_vehicle_type(3) == "Это трицикл марки Can-Am."

vehicle3 = Vehicle("Toyota", 4)
assert vehicle3.get_vehicle_type(4) == "Это автомобиль марки Toyota."

vehicle4 = Vehicle("Unknown", 6)
assert vehicle4.get_vehicle_type(6) == "Я не знаю таких ТС."

def test_get_vehicle_advice(): vehicle1 = Vehicle("Honda", 40000) assert vehicle1.get_vehicle_advice() == "Неплохо Honda можно брать."

vehicle2 = Vehicle("Ford", 75000)
assert vehicle2.get_vehicle_advice() == "Ford надо внимательно проверить."

vehicle3 = Vehicle("Chevrolet", 120000)
assert vehicle3.get_vehicle_advice() == "Chevrolet надо провести полную диагностику."

vehicle4 = Vehicle("BMW", 200000)
assert vehicle4.get_vehicle_advice() == "BMW лучше не покупать."

if name == "main": pytest.main()