Skip to content

Commit

Permalink
cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
swryan committed Jun 14, 2019
1 parent 01463ef commit db16fc0
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions openmdao/core/tests/test_distrib_list_vars.py
Expand Up @@ -306,8 +306,7 @@ def test_parallel_list_vars(self):

for i in range(len(expected)):
self.assertTrue(text[i].startswith(expected[i]),
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))
self.assertEqual(1, text.count("6 Input(s) in 'model'"))
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))

#
# list inputs, hierarchical
Expand Down Expand Up @@ -342,14 +341,13 @@ def test_parallel_list_vars(self):

for i in range(len(expected)):
self.assertTrue(text[i].startswith(expected[i]),
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))

#
# list outputs, not hierarchical
#
stream = cStringIO()
prob.model.list_outputs(values=True, residuals=True, hierarchical=False, out_stream=stream)
text = stream.getvalue()

if prob.comm.rank == 0:
text = stream.getvalue().split('\n')
Expand All @@ -375,7 +373,7 @@ def test_parallel_list_vars(self):

for i in range(len(expected)):
self.assertTrue(text[i].startswith(expected[i]),
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))

#
# list outputs, hierarchical
Expand Down Expand Up @@ -416,7 +414,7 @@ def test_parallel_list_vars(self):

for i in range(len(expected)):
self.assertTrue(text[i].startswith(expected[i]),
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))
'\nExpected: %s\nReceived: %s\n' % (expected[i], text[i]))


if __name__ == "__main__":
Expand Down

0 comments on commit db16fc0

Please sign in to comment.