Commit e1163159 authored by Michael Salim's avatar Michael Salim
Browse files

changed test_runners to work on Theta

parent 483ff54e
......@@ -61,7 +61,8 @@ class TestMPIRunner(BalsamTestCase):
'''specific check of mock_mpi_app.py output'''
found = []
for line in fp:
found.append(int(line.split()[1]))
if line.startswith('Rank'):
found.append(int(line.split()[1]))
self.assertSetEqual(set(range(n)), set(found))
def test_normal(self):
......@@ -89,6 +90,7 @@ class TestMPIRunner(BalsamTestCase):
# Check that the correct output is really there:
outpath = runner.outfile.name
with open(outpath) as fp:
self.assert_output_file_contains_n_ranks(fp, num_nodes*rpn)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment