File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -253,10 +253,16 @@ def add_args(new_data: dict[str, Any]) -> bool:
253
253
if generators_path .exists ():
254
254
generators_data = read_yaml (generators_path )
255
255
assert generators_data is not None
256
- assert isinstance (generators_data , dict )
256
+ assert isinstance (generators_data , CommentedMap )
257
257
258
258
if "testdata.yaml" not in generators_data :
259
- generators_data ["testdata.yaml" ] = CommentedMap ()
259
+ if "data" in generators_data :
260
+ # insert before data
261
+ pos = [* generators_data .keys ()].index ("data" )
262
+ generators_data .insert (pos , "testdata.yaml" , CommentedMap ())
263
+ else :
264
+ # insert at end
265
+ generators_data ["testdata.yaml" ] = CommentedMap ()
260
266
if add_args (generators_data ["testdata.yaml" ]):
261
267
write_yaml (generators_data , generators_path )
262
268
else :
You can’t perform that action at this time.
0 commit comments