Skip to content

Commit

Permalink
more tests in test_service_group_handle.py
Browse files Browse the repository at this point in the history
  • Loading branch information
mdorier committed Jan 31, 2024
1 parent 10eb477 commit 0e4b8fc
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions python/mochi/bedrock/test_service_group_handle.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,5 +98,16 @@ def test_spec(self):
self.assertIn(self_address, s)
self.assertIsInstance(s[self_address], spec.ProcSpec)

def test_query(self):
script = "return $__config__.margo.argobots;"
result = self.sgh.query(script)
self.assertIsInstance(result, dict)
self.assertEqual(len(result.keys()), 1)
self_address = str(self.server.margo.engine.address)
self.assertIn(self_address, result)
# check that we can convert the result into an ArgobotsSpec
s = spec.ArgobotsSpec.from_dict(result[self_address])


if __name__ == '__main__':
unittest.main()

0 comments on commit 0e4b8fc

Please sign in to comment.