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)