def setUp(self): self.sb = MainBearing() self.sb.lss_design_torque = 18691533.1165 self.sb.lss_diameter = 1.2115 self.sb.lss_mass = 42381.5447 self.sb.rotor_speed = 12.1 self.sb.rotor_diameter = 126.0
class Test_SecondBearing(unittest.TestCase): def setUp(self): self.sb = MainBearing() self.sb.lss_design_torque = 18691533.1165 self.sb.lss_diameter = 1.2115 self.sb.lss_mass = 42381.5447 self.sb.rotor_speed = 12.1 self.sb.rotor_diameter = 126.0 def test_functionality(self): self.sb.run() self.assertEqual(round(self.sb.mass, 1), 7348.5) def test_gradient(self): check_gradient_unit_test(self, self.sb, display=False)