update prediction service tests

src/allencell_ml_segmenter/_tests/fakes/fake_viewer.py

@@ -10,9 +10,10 @@
 
 
 class FakeLayer:
-    def __init__(self):
+    def __init__(self, name=""):
         self.removed: List[Layer] = []
         self.connected: List[Callable] = []
+        self.name = name
 
     def remove(self, layer_remove: Layer):
         self.removed.append(layer_remove)
@@ -66,3 +67,7 @@ def subscribe_layers_change_event(self, function):
 
     def get_layers(self):
         return self.images_added.keys()
+
+    def add_layer(self, name):
+        self.layer_list.append(FakeLayer(name))
+

src/allencell_ml_segmenter/_tests/services/test_prediction_service.py

@@ -156,6 +156,8 @@ def test_build_overrides() -> None:
         / "prediction_output_test"
     )
     assert overrides["data.transforms.predict.transforms[0].reader[0].C"] == 3
+    assert overrides["data.columns"] == ["raw", "split"]
+    assert overrides["data.split_column"] == "split"
 
 
 def test_build_overrides_experiment_none() -> None: