Update tests (ref #593)

fractal-analytics-platform · Nov 27, 2023 · 78461d0 · 78461d0
1 parent 7043d64
commit 78461d0
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/tests/tasks/test_workflows_napari_workflows.py b/tests/tasks/test_workflows_napari_workflows.py
@@ -92,6 +92,7 @@ def test_napari_workflow(
     output_specs = {
         "regionprops_DAPI": {  # type: ignore # noqa
             "type": "dataframe",
+            "label_name": "label_DAPI",
             "table_name": "regionprops_DAPI",
         },
     }
@@ -233,7 +234,11 @@ def test_napari_worfklow_label_input_only(
     input_label={"type": "label", "label_name": LABEL_NAME},
 )
 output_specs = dict(
-    output_dataframe={"type": "dataframe", "table_name": TABLE_NAME}
+    output_dataframe={
+        "type": "dataframe",
+        "table_name": TABLE_NAME,
+        "label_name": LABEL_NAME,
+    }
 )
 RELABELING_CASE_2: list = [workflow_file_name, input_specs, output_specs]
 # 3. Mixed labeling/measurement workflow.
@@ -243,7 +248,11 @@ def test_napari_worfklow_label_input_only(
 )
 output_specs = dict(
     output_label={"type": "label", "label_name": LABEL_NAME},
-    output_dataframe={"type": "dataframe", "table_name": TABLE_NAME},
+    output_dataframe={
+        "type": "dataframe",
+        "table_name": TABLE_NAME,
+        "label_name": LABEL_NAME,
+    },
 )
 RELABELING_CASE_3: list = [workflow_file_name, input_specs, output_specs]
 # Assemble three cases
@@ -516,6 +525,7 @@ def test_napari_workflow_empty_input_ROI_table(
         "regionprops_DAPI": {  # type: ignore # noqa
             "type": "dataframe",
             "table_name": "regionprops_DAPI",
+            "label_name": "label_DAPI",
         },
     }
     for component in metadata["image"]:
@@ -609,6 +619,7 @@ def test_napari_workflow_CYX(
         "regionprops_DAPI": {  # type: ignore # noqa
             "type": "dataframe",
             "table_name": "regionprops_DAPI",
+            "label_name": "label_DAPI",
         },
     }
     for component in metadata["image"]:
@@ -789,6 +800,7 @@ def test_napari_workflow_mock(
         "regionprops_DAPI": {  # type: ignore # noqa
             "type": "dataframe",
             "table_name": "regionprops_DAPI",
+            "label_name": "label_DAPI",
         },
     }
 

diff --git a/tests/test_unit_input_models.py b/tests/test_unit_input_models.py
@@ -98,6 +98,13 @@ def test_NapariWorkflowsOutput():
         )
     debug(e.value)
 
+    with pytest.raises(ValueError) as e:
+        NapariWorkflowsOutput(
+            type="dataframe",
+            table_name="something",
+        )
+    debug(e.value)
+
     # Valid
 
     specs = NapariWorkflowsOutput(
@@ -110,6 +117,7 @@ def test_NapariWorkflowsOutput():
 
     specs = NapariWorkflowsOutput(
         type="dataframe",
+        label_name="label_DAPI",
         table_name="table_DAPI",
     )
     debug(specs)