def test_add_hours(self):
""" Test whether the function add hours to exiting total hours """
d1 = Driver('Edison', 1.0, 1.000)
d1.add_hours(3.2)
self.assertEqual(4.2, d1.get_hours())
def test_get_hours(self):
""" Test whether the function returns the object's total hours """
d1 = Driver('Einstein', 0.0, 5.250)
self.assertEqual(5.25, d1.get_hours())