Skip to content

Commit d432a88

Browse files
committed
add preprocess
1 parent 220a4be commit d432a88

File tree

3 files changed

+15
-8
lines changed

3 files changed

+15
-8
lines changed

aviary/subsystems/geometry/flops_based/test/test_nacelle.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,9 @@ def test_case_BWB(self):
6060
prob = self.prob
6161

6262
options = get_flops_options('BWBsimpleFLOPS')
63-
# options = {
64-
# Aircraft.Engine.NUM_ENGINES: np.array([3]),
65-
# }
63+
options = {
64+
Aircraft.Engine.NUM_ENGINES: np.array([3]),
65+
}
6666

6767
prob.model.add_subsystem(
6868
'nacelles', Nacelles(), promotes_outputs=['*'], promotes_inputs=['*']

aviary/subsystems/mass/flops_based/test/test_air_conditioning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ def test_case(self, case_name):
185185
promotes_outputs=['*'],
186186
)
187187

188-
prob.model_options['*'] = get_flops_options(case_name)
188+
prob.model_options['*'] = get_flops_options(case_name, preprocess=True)
189189

190190
prob.setup(check=False, force_alloc_complex=True)
191191

aviary/subsystems/mass/flops_based/test/test_apu.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import unittest
22

3+
import numpy as np
34
import openmdao.api as om
45
from openmdao.utils.assert_utils import assert_check_partials
56
from openmdao.utils.testing_utils import use_tempdirs
@@ -88,8 +89,11 @@ class BWBAPUMassTest(unittest.TestCase):
8889
def setUp(self):
8990
self.prob = om.Problem()
9091

91-
@parameterized.expand(get_flops_case_names(only=bwb_cases), name_func=print_case)
92-
def test_case(self, case_name):
92+
# @parameterized.expand(get_flops_case_names(only=bwb_cases), name_func=print_case)
93+
def test_case(
94+
self,
95+
):
96+
case_name = 'BWBsimpleFLOPS'
9397
prob = self.prob
9498

9599
prob.model.add_subsystem(
@@ -99,7 +103,7 @@ def test_case(self, case_name):
99103
promotes_outputs=['*'],
100104
)
101105

102-
prob.model_options['*'] = get_flops_options(case_name, preprocess=False)
106+
prob.model_options['*'] = get_flops_options(case_name, preprocess=True)
103107

104108
prob.setup(check=False, force_alloc_complex=True)
105109

@@ -114,4 +118,7 @@ def test_case(self, case_name):
114118

115119

116120
if __name__ == '__main__':
117-
unittest.main()
121+
# unittest.main()
122+
test = BWBAPUMassTest()
123+
test.setUp()
124+
test.test_case()

0 commit comments

Comments
 (0)