fixing test

mochi-hpc · Jan 31, 2024 · 8151d76 · 8151d76
1 parent 78d806a
commit 8151d76
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/python/mochi/bedrock/test_service_handle.py b/python/mochi/bedrock/test_service_handle.py
@@ -173,14 +173,7 @@ def test_add_xstream_from_spec(self):
         self.add_xstream(proc.margo.argobots.xstreams["my_xstream"])
 
     def test_remove_xstream(self):
-        config = {
-            "name": "my_xstream",
-            "scheduler": {
-                "type": "basic_wait",
-                "pools": ["__primary__"]
-            }
-        }
-        self.add_xstream(config)
+        self.test_add_xstream_from_dict()
         initial_num_xstreams = len(self.server.margo.xstreams)
         self.sh.remove_xstream("my_xstream")
         self.assertEqual(len(self.server.margo.xstreams), initial_num_xstreams - 1)