1+ from collections .abc import Generator
12import contextlib
23from _pytest import monkeypatch
34import numpy
@@ -138,9 +139,11 @@ def prevent_script_exit(monkeypatch: monkeypatch.MonkeyPatch) -> None:
138139
139140
140141@pytest .fixture
141- def create_test_run (request , prevent_script_exit ) -> typing .Generator [typing .Tuple [sv_run .Run , dict ], None , None ]:
142+ def create_test_run (request , prevent_script_exit ) -> Generator [tuple [sv_run .Run , dict ]]:
143+ _ = prevent_script_exit
142144 with sv_run .Run () as run :
143- _test_run_data = setup_test_run (run , True , request )
145+ with tempfile .TemporaryDirectory () as tempd :
146+ _test_run_data = setup_test_run (run , temp_dir = pathlib .Path (tempd ), create_objects = True , request = request )
144147 yield run , _test_run_data
145148 with contextlib .suppress (ObjectNotFoundError ):
146149 sv_api_obj .Folder (identifier = run ._folder .id ).delete (recursive = True , delete_runs = True , runs_only = False )
@@ -151,13 +154,12 @@ def create_test_run(request, prevent_script_exit) -> typing.Generator[typing.Tup
151154
152155
153156@pytest .fixture
154- def create_test_run_offline (request , monkeypatch : pytest .MonkeyPatch , prevent_script_exit ) -> typing .Generator [typing .Tuple [sv_run .Run , dict ], None , None ]:
155- def testing_exit (status : int ) -> None :
156- raise SystemExit (status )
157+ def create_test_run_offline (request , monkeypatch : pytest .MonkeyPatch , prevent_script_exit ) -> Generator [tuple [sv_run .Run , dict ]]:
158+ _ = prevent_script_exit
157159 with tempfile .TemporaryDirectory () as temp_d :
158160 monkeypatch .setenv ("SIMVUE_OFFLINE_DIRECTORY" , temp_d )
159161 with sv_run .Run ("offline" ) as run :
160- _test_run_data = setup_test_run (run , temp_d , True , request )
162+ _test_run_data = setup_test_run (run , temp_dir = pathlib . Path ( temp_d ), create_objects = True , request = request )
161163 yield run , _test_run_data
162164 with contextlib .suppress (ObjectNotFoundError ):
163165 sv_api_obj .Folder (identifier = run ._folder .id ).delete (recursive = True , delete_runs = True , runs_only = False )
@@ -168,29 +170,34 @@ def testing_exit(status: int) -> None:
168170
169171
170172@pytest .fixture
171- def create_plain_run (request , prevent_script_exit , mocker ) -> typing .Generator [typing .Tuple [sv_run .Run , dict ], None , None ]:
173+ def create_plain_run (request , prevent_script_exit , mocker ) -> Generator [tuple [sv_run .Run , dict ]]:
174+ _ = prevent_script_exit
172175 def testing_exit (status : int ) -> None :
173176 raise SystemExit (status )
174177 with sv_run .Run () as run :
175178 run .metric_spy = mocker .spy (run , "_get_internal_metrics" )
176- yield run , setup_test_run (run , False , request )
179+ with tempfile .TemporaryDirectory () as tempd :
180+ yield run , setup_test_run (run , temp_dir = pathlib .Path (tempd ), create_objects = False , request = request )
177181 clear_out_files ()
178182
179183
180184@pytest .fixture
181- def create_pending_run (request , prevent_script_exit ) -> typing .Generator [typing .Tuple [sv_run .Run , dict ], None , None ]:
185+ def create_pending_run (request , prevent_script_exit ) -> Generator [tuple [sv_run .Run , dict ]]:
186+ _ = prevent_script_exit
182187 with sv_run .Run () as run :
183- yield run , setup_test_run (run , False , request , True )
188+ with tempfile .TemporaryDirectory () as tempd :
189+ yield run , setup_test_run (run , temp_dir = pathlib .Path (tempd ), create_objects = False , request = request , created_only = True )
184190 clear_out_files ()
185191
186192
187193@pytest .fixture
188- def create_plain_run_offline (request ,prevent_script_exit ,monkeypatch ) -> typing .Generator [typing .Tuple [sv_run .Run , dict ], None , None ]:
194+ def create_plain_run_offline (request ,prevent_script_exit ,monkeypatch ) -> Generator [tuple [sv_run .Run , dict ]]:
195+ _ = prevent_script_exit
189196 with tempfile .TemporaryDirectory () as temp_d :
190197 monkeypatch .setenv ("SIMVUE_OFFLINE_DIRECTORY" , temp_d )
191198 with sv_run .Run ("offline" ) as run :
192199 _temporary_directory = pathlib .Path (temp_d )
193- yield run , setup_test_run (run , _temporary_directory ,False , request )
200+ yield run , setup_test_run (run , temp_dir = _temporary_directory , create_objects = False , request = request )
194201 clear_out_files ()
195202
196203
@@ -207,7 +214,7 @@ def testing_exit(status: int) -> None:
207214 _folder .delete (recursive = True , runs_only = False , delete_runs = True )
208215
209216
210- def setup_test_run (run : sv_run .Run , temp_dir : pathlib .Path , create_objects : bool , request : pytest .FixtureRequest , created_only : bool = False ):
217+ def setup_test_run (run : sv_run .Run , * , temp_dir : pathlib .Path , create_objects : bool , request : pytest .FixtureRequest , created_only : bool = False ):
211218 fix_use_id : str = str (uuid .uuid4 ()).split ('-' , 1 )[0 ]
212219 _test_name : str = request .node .name .replace ("[" , "_" ).replace ("]" , "" )
213220 TEST_DATA = {
0 commit comments