def setUp(self): self.hub = HubSystemCostAdder() self.hub.hub_cost = 20000.0 self.hub.pitch_system_cost = 20000.0 self.hub.spinner_cost = 20000.0
def test1(self): hub = HubSystemCostAdder() hub.hub_cost = 20000.0 hub.pitch_system_cost = 20000.0 hub.spinner_cost = 20000.0 check_gradient_unit_test(self, hub)
class TestHubSystemCostAdder(unittest.TestCase): def setUp(self): self.hub = HubSystemCostAdder() self.hub.hub_cost = 20000.0 self.hub.pitch_system_cost = 20000.0 self.hub.spinner_cost = 20000.0 def test_functionality(self): self.hub.run() self.assertEqual(round(self.hub.cost,2), 60000.00) def test_gradient(self): check_gradient_unit_test(self, self.hub)