6

tests/test_loop_q3.py

@@ -22,9 +22,10 @@ from mcm26b.coverage import coverage_points
 from mcm26b.evaluation import recompute_virtual_time
 from mcm26b.planner import OmnidirectionalLoop
 from mcm26b.protocol.client import SimulatorClient
-from mcm26b.sim import (LocalSimTransport, LocalSimulator,
-                        ScriptedFaultTransport, Source)
-from mcm26b.sim.scenes import random_omni_scene
+from mcm26b.harness import (LocalSimTransport, LocalSimulator,
+                            ScriptedFaultTransport)
+from referee import Source
+from mcm26b.harness import random_omni_scene
 
 CONFIG = replace(ProblemConfig(),
                  search=replace(ProblemConfig().search,
@@ -216,7 +217,7 @@ class SimulatorVirtualCapTests(unittest.TestCase):
         simulator = LocalSimulator([], robot_id="T001", config=CONFIG)
         simulator.entered = True
         simulator.virtual_time_s = CONFIG.field.max_virtual_duration_s
-        from mcm26b.sim import SimulatorClosed
+        from referee import SimulatorClosed
         with self.assertRaises(SimulatorClosed):
             simulator.handle(
                 "/measure", {"arena_id": "default", "robot_id": "T001",