rewriting the tests for registration connector

Author: birajstha

CPAC/registration/tests/test_ants_registration_connector.py

@@ -1,25 +1,25 @@
 import pytest
+
 from CPAC.registration.registration import ANTs_registration_connector
 
-@pytest.mark.parametrize("sink_native_transforms", ['On', 'Off'])
+
+@pytest.mark.parametrize("sink_native_transforms", ["On", "Off"])
 def test_ants_registration_connector(sink_native_transforms):
     cfg = {
-        'registration-workflows': {'sink_native_transforms': sink_native_transforms},
+        "registration-workflows": {"sink_native_transforms": sink_native_transforms},
     }
-    params = {'metric': 'MI'}
-    _, outputs = ANTs_registration_connector(
-        wf_name='test', cfg=cfg, params=params
-    )
+    params = {"metric": "MI"}
+    _, outputs = ANTs_registration_connector(wf_name="test", cfg=cfg, params=params)
     expected_keys = {
-        'from-T1w_to-template_mode-image_desc-initial_xfm',
-        'from-T1w_to-template_mode-image_desc-rigid_xfm',
-        'from-T1w_to-template_mode-image_desc-affine_xfm',
+        "from-T1w_to-template_mode-image_desc-initial_xfm",
+        "from-T1w_to-template_mode-image_desc-rigid_xfm",
+        "from-T1w_to-template_mode-image_desc-affine_xfm",
     }
     if sink_native_transforms:
-        assert expected_keys.issubset(outputs.keys()), (
-            f"Expected outputs {expected_keys} not found in {outputs.keys()}"
-        )
+        assert expected_keys.issubset(
+            outputs.keys()
+        ), f"Expected outputs {expected_keys} not found in {outputs.keys()}"
     else:
-        assert not expected_keys.intersection(outputs.keys()), (
-            f"Outputs {expected_keys} should not be present when sink_native_transforms is Off"
-        )
+        assert not expected_keys.intersection(
+            outputs.keys()
+        ), f"Outputs {expected_keys} should not be present when sink_native_transforms is Off"

CPAC/registration/tests/test_fsl_registration_connector.py

@@ -1,27 +1,29 @@
 import pytest
+
 from CPAC.registration.registration import FSL_registration_connector
 
-@pytest.mark.parametrize("sink_native_transforms", ['On', 'Off'])
+
+@pytest.mark.parametrize("sink_native_transforms", ["On", "Off"])
 def test_fsl_registration_connector(sink_native_transforms):
-    wf_name = 'test_fsl_registration_connector'
+    wf_name = "test_fsl_registration_connector"
     cfg = {
-        'registration-workflows': {'sink_native_transforms': sink_native_transforms},
+        "registration-workflows": {"sink_native_transforms": sink_native_transforms},
     }
     _, outputs = FSL_registration_connector(wf_name, cfg)
-    if sink_native_transforms == 'On':
+    if sink_native_transforms == "On":
         expected_outputs = {
-            'from-T1w_to-template_mode-image_desc-linear_xfm',
-            'from-template_to-T1w_mode-image_desc-linear_xfm'
+            "from-T1w_to-template_mode-image_desc-linear_xfm",
+            "from-template_to-T1w_mode-image_desc-linear_xfm",
         }
-        assert expected_outputs.issubset(outputs.keys()), (
-            f"Expected outputs {expected_outputs} not found in {outputs.keys()}"
-        )
+        assert expected_outputs.issubset(
+            outputs.keys()
+        ), f"Expected outputs {expected_outputs} not found in {outputs.keys()}"
     else:
         # Adjust this set based on what outputs should be present when 'Off'
         not_expected_outputs = {
-            'from-T1w_to-template_mode-image_desc-linear_xfm',
-            'from-template_to-T1w_mode-image_desc-linear_xfm'
+            "from-T1w_to-template_mode-image_desc-linear_xfm",
+            "from-template_to-T1w_mode-image_desc-linear_xfm",
         }
-        assert not not_expected_outputs.intersection(outputs.keys()), (
-            f"Outputs {not_expected_outputs} should not be present when sink_native_transforms is Off"
-        )
+        assert not not_expected_outputs.intersection(
+            outputs.keys()
+        ), f"Outputs {not_expected_outputs} should not be present when sink_native_transforms is Off"