class TestPitchSystemCost(unittest.TestCase): def setUp(self): self.pitch = PitchSystemCost() self.pitch.pitch_system_mass = 17004.0 self.pitch.year = 2009 self.pitch.month = 12 def test_functionality(self): self.pitch.run() self.assertEqual(round(self.pitch.cost,2), 535075.83) def test_gradient(self): check_gradient_unit_test(self, self.pitch)