def setUp(self): self.bearings = BearingsCost() self.bearings.main_bearing_mass = 9731.41 / 2.0 self.bearings.second_bearing_mass = 9731.41 / 2.0 self.bearings.year = 2009 self.bearings.month = 12
def test1(self): bearings = BearingsCost() bearings.main_bearing_mass = 9731.41 / 2.0 bearings.second_bearing_mass = 9731.41 / 2.0 bearings.year = 2009 bearings.month = 12 check_gradient_unit_test(self, bearings)
class TestBearingsCost(unittest.TestCase): def setUp(self): self.bearings = BearingsCost() self.bearings.main_bearing_mass = 9731.41 / 2.0 self.bearings.second_bearing_mass = 9731.41 / 2.0 self.bearings.year = 2009 self.bearings.month = 12 def test_functionality(self): self.bearings.run() self.assertEqual(round(self.bearings.cost,2), 56660.73) def test_gradient(self): check_gradient_unit_test(self, self.bearings)