Source code for niworkflows.utils.testing

-from copy import deepcopy
-import json
+import json
+from copy import deepcopy
 from pathlib import Path
+
 import yaml
 
 
@@ -360,38 +361,37 @@ 
Source code for niworkflows.utils.testing
         try:
             bids_dict = json.loads(bids_config)
         except json.JSONDecodeError:
-            bids_dict = yaml.load(bids_config, Loader=yaml.Loader)
+            bids_dict = yaml.safe_load(bids_config)
 
     _bids_dict = deepcopy(bids_dict)
     root = Path(target_path).absolute()
     root.mkdir(parents=True)
 
-    desc = bids_dict.pop("dataset_description", None)
+    desc = bids_dict.pop('dataset_description', None)
     if desc is None:
         # default description
-        desc = {"Name": "Default", "BIDSVersion": "1.6.0"}
-    to_json(root / "dataset_description.json", desc)
+        desc = {'Name': 'Default', 'BIDSVersion': '1.6.0'}
+    to_json(root / 'dataset_description.json', desc)
 
     cached_subject_data = None
     for subject, sessions in bids_dict.items():
-        bids_subject = subject if subject.startswith("sub-") else f"sub-{subject}"
+        bids_subject = subject if subject.startswith('sub-') else f'sub-{subject}'
         subj_path = root / bids_subject
         subj_path.mkdir(exist_ok=True)
 
-        if sessions == "*":  # special case to copy previous subject data
+        if sessions == '*':  # special case to copy previous subject data
             sessions = cached_subject_data.copy()
 
         if isinstance(sessions, dict):  # single session
-            sessions.update({"session": None})
+            sessions.update({'session': None})
             sessions = [sessions]
 
         cached_subject_data = deepcopy(sessions)
         for session in sessions:
-
-            ses_name = session.pop("session", None)
+            ses_name = session.pop('session', None)
             if ses_name is not None:
-                bids_session = ses_name if ses_name.startswith("ses-") else f"ses-{ses_name}"
-                bids_prefix = f"{bids_subject}_{bids_session}"
+                bids_session = ses_name if ses_name.startswith('ses-') else f'ses-{ses_name}'
+                bids_prefix = f'{bids_subject}_{bids_session}'
                 curr_path = subj_path / bids_session
                 curr_path.mkdir(exist_ok=True)
             else:
@@ -407,16 +407,16 @@ 
Source code for niworkflows.utils.testing
                     files = [files]
 
                 for bids_file in files:
-                    metadata = bids_file.pop("metadata", None)
-                    extension = bids_file.pop("extension", ".nii.gz")
-                    suffix = bids_file.pop("suffix")
+                    metadata = bids_file.pop('metadata', None)
+                    extension = bids_file.pop('extension', '.nii.gz')
+                    suffix = bids_file.pop('suffix')
                     entities = combine_entities(**bids_file)
-                    data_file = modality_path / f"{bids_prefix}{entities}_{suffix}{extension}"
+                    data_file = modality_path / f'{bids_prefix}{entities}_{suffix}{extension}'
                     data_file.touch()
 
                     if metadata is not None:
                         out_metadata = data_file.parent / data_file.name.replace(
-                            extension, ".json"
+                            extension, '.json'
                         )
                         to_json(out_metadata, metadata)
 
@@ -436,7 +436,7 @@ 
Source code for niworkflows.utils.testing
 

 [docs]
 def combine_entities(**entities):
-    return f"_{'_'.join([f'{lab}-{val}' for lab, val in entities.items()])}" if entities else ""

+    return f"_{'_'.join([f'{lab}-{val}' for lab, val in entities.items()])}" if entities else ''