class TestPetrolCar(unittest.TestCase):
def setUp(self):
# The PetrolCar class calls itself.
self.petrol_car = PetrolCar()
def test_PetrolCar_fuel_cylinders(self):
# Testing a PetrolCar the default value of 3 is called from the base class Petrol Car and is tested to
# ensure the base class PetrolCar functions are being correctly implemented. The setNumberOfFuelCylinders
# in this case changes the default value from 3 to 7 this is tested and it has become 7 so is true.
# This is similar to ElectricCar FuelCells implementation but with cylinders instead of cells.
# The cylinders are tested with values for 2 and 5 which pass.
self.assertEqual(3, self.petrol_car.getNumberOfFuelCylinders())
self.petrol_car.setNumberOfFuelCylinders(7)
self.assertEqual(7, self.petrol_car.getNumberOfFuelCylinders())
self.petrol_car.setNumberOfFuelCylinders(2)
self.assertEqual(2, self.petrol_car.getNumberOfFuelCylinders())
self.petrol_car.setNumberOfFuelCylinders(5)
self.assertEqual(5, self.petrol_car.getNumberOfFuelCylinders())
def test_PetrolCar_mileage_and_move(self):
# Ths function tests the mileage of the PetrolCar class. As in the other tests it starts with the default
# value from the base class. Move is then called using the value 5 this is tested against the value
# retrieved from getMileage which is now increased to 5 so it's true. It is increased by 12 to make 17,
# tested by adding 0 still gives 17, tested adding a decimal 6.5 to get 23.5.
self.assertEqual(0, self.petrol_car.getMileage())
self.petrol_car.move(5)
self.assertEqual(5, self.petrol_car.getMileage())
self.petrol_car.move(12)
self.assertEqual(17, self.petrol_car.getMileage())
self.petrol_car.move(0)
self.assertEqual(17, self.petrol_car.getMileage())
self.petrol_car.move(6.5)
self.assertEqual(23.5, self.petrol_car.getMileage())
def test_PetrolCar_make(self):
# Testing make function for PetrolCar to ensure correctly inherited from base class Car.
self.assertEqual(' ', self.petrol_car.getMake())
self.petrol_car.setMake('Ford')
self.assertEqual('Ford', self.petrol_car.getMake())
self.petrol_car.setMake('Ford Focus')
self.assertEqual('Ford Focus', self.petrol_car.getMake())
def test_PetrolCar_engineSize(self):
# Testing engineSize for PetrolCar class.
self.assertEqual(' ', self.petrol_car.getEngineSize())
self.petrol_car.setEngineSize(2.2)
self.assertEqual(2.2, self.petrol_car.getEngineSize())
self.petrol_car.setEngineSize(2.6)
self.assertEqual(2.6, self.petrol_car.getEngineSize())
def test_PetrolCar_colour_and_paint(self):
# Testing colour and paint for PetrolCar Class
self.assertEqual(' ', self.petrol_car.getColour())
self.petrol_car.paint('black')
self.assertEqual('black', self.petrol_car.getColour())
self.petrol_car.setColour('dark blue')
self.assertEqual('dark blue', self.petrol_car.getColour())
car3.setNumberOfFuelCylinders(1)
print('The number of fuel cylinders is ' + str(car3.getNumberOfFuelCylinders()))
# Car4 calls the PetrolCar class and performs similar functions as other class'. The colour is red, mileage
# is set to 150 and increased by 25 to 175kms. The engine size is set at 2.9. The petrol car is a Toyota Corolla
# and has 3 cylinders set as the value for the variable NumberOfFuelCylinders.
car4 = PetrolCar()
car4.setMake('Toyota Corolla')
print '\nOur petrol option is ' + car4.getMake()
car4.setColour('Red')
print('The colour is: ' + car4.getColour())
car4.setMileage(150)
car4.move(25)
print('The mileage is ' + str(car4.getMileage())) + 'kms'
car4.engineSize = '2.9'
print 'The engine size is ' + car4.engineSize
car4.setNumberOfFuelCylinders(3)
print('The number of fuel cylinders is ' + str(car4.getNumberOfFuelCylinders()))
# Car5 is calling a DieselCar objects and will implement its functions and variables. Similar to other class'
# the car sets a model and returns it in this case Mercedes Benz, the colour is set as silver, the mileage is
# 1000 and increased by 40 to 1040 as the mileage returned, the engineSize is 3.2 and the car uses 5 cylinders.
car5 = DieselCar()
car5.setMake('Mercedes Benz')
print '\nOur diesel option is ' + car4.getMake()