Написать пайтест для класса: 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 unittest from vehicle import Vehicle

class TestVehicle(unittest.TestCase):

def setUp(self):
    self.vehicle1 = Vehicle("Honda", 30000)
    self.vehicle2 = Vehicle("Yamaha", 60000)
    self.vehicle3 = Vehicle("Toyota", 120000)
    self.vehicle4 = Vehicle("BMW", 200000)

def test_get_vehicle_type(self):
    self.assertEqual(self.vehicle1.get_vehicle_type(2), "Это мотоцикл марки Honda.")
    self.assertEqual(self.vehicle2.get_vehicle_type(3), "Это трицикл марки Yamaha.")
    self.assertEqual(self.vehicle3.get_vehicle_type(4), "Это автомобиль марки Toyota.")
    self.assertEqual(self.vehicle4.get_vehicle_type(5), "Я не знаю таких ТС.")

def test_get_vehicle_advice(self):
    self.assertEqual(self.vehicle1.get_vehicle_advice(), "Неплохо Honda можно брать.")
    self.assertEqual(self.vehicle2.get_vehicle_advice(), "Yamaha надо внимательно проверить.")
    self.assertEqual(self.vehicle3.get_vehicle_advice(), "Toyota надо провести полную диагностику.")
    self.assertEqual(self.vehicle4.get_vehicle_advice(), "BMW лучше не покупать.")

if name == 'main': unittest.main()