def setUp(self):

        self.generator = GeneratorCost()

        self.generator.generator_mass = 16699.85
        self.generator.year = 2009
        self.generator.month = 12
        self.generator.drivetrain_design = 'geared'
        self.generator.machine_rating = 5000.0
class TestGeneratorCost(unittest.TestCase):

    def setUp(self):

        self.generator = GeneratorCost()

        self.generator.generator_mass = 16699.85
        self.generator.year = 2009
        self.generator.month = 12
        self.generator.drivetrain_design = 'geared'
        self.generator.machine_rating = 5000.0

    def test_functionality(self):
    
        self.generator.run()
        
        self.assertEqual(round(self.generator.cost,2), 435157.71)

    def test_gradient(self):

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

        generator = GeneratorCost()

        generator.generator_mass = 16699.85
        generator.year = 2009
        generator.month = 12
        generator.drivetrain_design = 'geared'
        generator.machine_rating = 5000.0

        check_gradient_unit_test(self, generator)
    def test1(self):

        generator = GeneratorCost()

        generator.generator_mass = 16699.85
        generator.year = 2009
        generator.month = 12
        generator.drivetrain_design = "geared"
        generator.machine_rating = 5000.0

        check_gradient_unit_test(self, generator)