@@ -354,7 +354,7 @@ def test_flag_large_events_withsnowball_noextension():
354
354
assert cube [0 , 2 , 2 , 2 ] == 0 # Saturation was NOT extended due to max_extended_radius=1
355
355
356
356
357
- def test_find_faint_extended ():
357
+ def test_find_faint_extended (tmp_path ):
358
358
nint , ngrps , ncols , nrows = 1 , 66 , 25 , 25
359
359
data = np .zeros (shape = (nint , ngrps , nrows , ncols ), dtype = np .float32 )
360
360
gdq = np .zeros_like (data , dtype = np .uint32 )
@@ -366,7 +366,7 @@ def test_find_faint_extended():
366
366
rng = np .random .default_rng (12345 )
367
367
data [0 , 1 :, 14 :20 , 15 :20 ] = 6 * gain * 6.0 * np .sqrt (2 )
368
368
data = data + rng .normal (size = (nint , ngrps , nrows , ncols )) * readnoise
369
- fits .writeto ("data.fits" , data , overwrite = True )
369
+ fits .writeto (tmp_path / "data.fits" , data , overwrite = True )
370
370
gdq , num_showers = find_faint_extended (
371
371
data ,
372
372
gdq ,
@@ -386,7 +386,7 @@ def test_find_faint_extended():
386
386
)
387
387
# Check that all the expected samples in group 2 are flagged as jump and
388
388
# that they are not flagged outside
389
- fits .writeto ("gdq.fits" , gdq , overwrite = True )
389
+ fits .writeto (tmp_path / "gdq.fits" , gdq , overwrite = True )
390
390
# assert num_showers == 1
391
391
assert np .all (gdq [0 , 1 , 22 , 14 :23 ] == 0 )
392
392
assert gdq [0 , 1 , 16 , 18 ] == DQFLAGS ['JUMP_DET' ]
0 commit comments