Apply codeformat fixes and tighten validation

Signed-off-by: Zixun Wang <craddywang@gmail.com>

Author: ZixunWang

monai/transforms/post/dictionary.py

@@ -196,7 +196,7 @@ def __init__(
         super().__init__(keys, allow_missing_keys)
         self.argmax = ensure_tuple_rep(argmax, len(self.keys))
         self.rankseg = ensure_tuple_rep(rankseg, len(self.keys))
-        if any(argmax_ and rankseg_ for argmax_, rankseg_ in zip(self.argmax, self.rankseg)):
+        if any(argmax_ and rankseg_ for argmax_, rankseg_ in zip(self.argmax, self.rankseg, strict=True)):
             raise ValueError("`rankseg=True` is incompatible with `argmax=True`.")
         self.to_onehot = []
         for flag in ensure_tuple_rep(to_onehot, len(self.keys)):

tests/transforms/test_as_discrete.py

@@ -67,23 +67,11 @@
     )
 
     TEST_CASES.append(
-        [
-            {"rankseg": False, "argmax": True},
-            p([[[0.3, 0.6]], [[0.7, 0.4]]]),
-            p([[[1.0, 0.0]]]),
-            (1, 1, 2),
-        ]
+        [{"rankseg": False, "argmax": True}, p([[[0.3, 0.6]], [[0.7, 0.4]]]), p([[[1.0, 0.0]]]), (1, 1, 2)]
     )
 
     if post_array.has_rankseg:
-        TEST_CASES.append(
-            [
-                {"rankseg": True},
-                p([[[0.3, 0.6]], [[0.7, 0.4]]]),
-                p([[[1.0, 1.0]]]),
-                (1, 1, 2),
-            ]
-        )
+        TEST_CASES.append([{"rankseg": True}, p([[[0.3, 0.6]], [[0.7, 0.4]]]), p([[[1.0, 1.0]]]), (1, 1, 2)])
 
 
 class TestAsDiscrete(unittest.TestCase):

tests/transforms/test_as_discreted.py

@@ -91,14 +91,8 @@
         TEST_CASES.append(
             [
                 {"keys": ["pred", "label"], "rankseg": [True, False]},
-                {
-                    "pred": p([[[0.3, 0.6]], [[0.7, 0.4]]]),
-                    "label": p([[[0.0, 1.0]]]),
-                },
-                {
-                    "pred": p([[[1.0, 1.0]]]),
-                    "label": p([[[0.0, 1.0]]]),
-                },
+                {"pred": p([[[0.3, 0.6]], [[0.7, 0.4]]]), "label": p([[[0.0, 1.0]]])},
+                {"pred": p([[[1.0, 1.0]]]), "label": p([[[0.0, 1.0]]])},
                 (1, 1, 2),
             ]
         )
@@ -116,9 +110,7 @@ def test_value_shape(self, input_param, test_input, output, expected_shape):
 
     def test_rankseg_argmax_incompatible(self):
         with self.assertRaises(ValueError):
-            AsDiscreted(keys="pred", argmax=True, rankseg=True)(
-                {"pred": [[[0.3, 0.6]], [[0.7, 0.4]]]}
-            )
+            AsDiscreted(keys="pred", argmax=True, rankseg=True)({"pred": [[[0.3, 0.6]], [[0.7, 0.4]]]})
 
     def test_rankseg_missing_dependency(self):
         with mock.patch("monai.transforms.post.array.has_rankseg", False):