Update tests/data/test_persistentdataset.py

Co-authored-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
Signed-off-by: Mason C. Cleveland <104479423+mccle@users.noreply.github.com>

Author: mccle

tests/data/test_persistentdataset.py

@@ -188,17 +188,9 @@ def test_track_meta_and_weights_only(self, track_meta, weights_only, expected_er
             transform = Compose([LoadImaged(keys=["image"])])
             cache_dir = os.path.join(os.path.join(tempdir, "cache"), "data")
 
-            if expected_error:
-                with self.assertRaises(ValueError):
-                    PersistentDataset(
-                        data=test_data,
-                        transform=transform,
-                        cache_dir=cache_dir,
-                        track_meta=track_meta,
-                        weights_only=weights_only,
-                    )
 
-            else:
+            cm = self.assertRaises(ValueError) if expected_error else contextlib.nullcontext()
+            with cm:
                 test_dataset = PersistentDataset(
                     data=test_data,
                     transform=transform,