6

tests/test_bayes_loop_q3.py

@@ -19,9 +19,10 @@ from mcm26b.config import ProblemConfig
 from mcm26b.evaluation import recompute_virtual_time
 from mcm26b.planner.bayes_loop import BayesLoop
 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,
@@ -147,8 +148,8 @@ class BayesLoopAcceptanceTests(unittest.TestCase):
             client = SimulatorClient("T001", config=CONFIG,
                                      transport=LocalSimTransport(simulator))
             loop = loop_cls(client, CONFIG)
-            expected = seven_points() if name in ("bayes", "seven") \
-                else coverage_points(CONFIG)
+            expected = seven_points(CONFIG.bayes.station_ring_radius_m) \
+                if name in ("bayes", "seven") else coverage_points(CONFIG)
             self.assertEqual(loop.ledger.points, expected)