def setUp(self):

        self.gearbox = GearboxCost()

        self.gearbox.gearbox_mass = 30237.60
        self.gearbox.year = 2009
        self.gearbox.month = 12
        self.gearbox.drivetrain_design = 'geared'
    def test1(self):

        gearbox = GearboxCost()

        gearbox.gearbox_mass = 30237.60
        gearbox.year = 2009
        gearbox.month = 12
        gearbox.drivetrain_design = "geared"

        check_gradient_unit_test(self, gearbox)
class TestGearboxCost(unittest.TestCase):

    def setUp(self):

        self.gearbox = GearboxCost()

        self.gearbox.gearbox_mass = 30237.60
        self.gearbox.year = 2009
        self.gearbox.month = 12
        self.gearbox.drivetrain_design = 'geared'

    def test_functionality(self):
    
        self.gearbox.run()
        
        self.assertEqual(round(self.gearbox.cost,2), 648030.64)

    def test_gradient(self):

        check_gradient_unit_test(self, self.gearbox)
    def test1(self):

        gearbox = GearboxCost()

        gearbox.gearbox_mass = 30237.60
        gearbox.year = 2009
        gearbox.month = 12
        gearbox.drivetrain_design = 'geared'

        check_gradient_unit_test(self, gearbox)