Update tests to use new fits API instead of removed output_to_fits methods

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: 2 people

test_autoarray/dataset/imaging/test_dataset.py

@@ -174,10 +174,14 @@ def test__output_to_fits__round_trips_data_psf_noise_map_correctly(
     imaging_7x7, test_data_path
 ):
 
-    imaging_7x7.output_to_fits(
+    from autoarray.dataset.plot.imaging_plots import fits_imaging
+
+    fits_imaging(
+        dataset=imaging_7x7,
         data_path=path.join(test_data_path, "data.fits"),
         psf_path=path.join(test_data_path, "psf.fits"),
         noise_map_path=path.join(test_data_path, "noise_map.fits"),
+        overwrite=True,
     )
 
     dataset = aa.Imaging.from_fits(

test_autoarray/dataset/interferometer/test_dataset.py

@@ -109,7 +109,10 @@ def test__output_all_arrays(mask_2d_7x7):
 
     os.makedirs(test_data_path)
 
-    dataset.output_to_fits(
+    from autoarray.dataset.plot.interferometer_plots import fits_interferometer
+
+    fits_interferometer(
+        dataset=dataset,
         data_path=path.join(test_data_path, "data.fits"),
         noise_map_path=path.join(test_data_path, "noise_map.fits"),
         uv_wavelengths_path=path.join(test_data_path, "uv_wavelengths.fits"),

test_autoarray/mask/test_mask_2d.py

@@ -392,7 +392,13 @@ def test__from_fits__output_to_fits__roundtrip_preserves_values_pixel_scales_and
 
     os.makedirs(output_path)
 
-    mask.output_to_fits(file_path=path.join(output_path, "mask.fits"))
+    from autoconf.fitsable import output_to_fits
+    output_to_fits(
+        values=mask.astype("float"),
+        file_path=path.join(output_path, "mask.fits"),
+        header_dict=mask.header_dict,
+        ext_name="mask",
+    )
 
     mask = aa.Mask2D.from_fits(
         file_path=path.join(output_path, "mask.fits"),

test_autoarray/structures/arrays/test_uniform_1d.py

@@ -166,7 +166,8 @@ def test__output_to_fits__ones_array__fits_file_has_correct_values_and_header():
 
     os.makedirs(test_data_path)
 
-    arr.output_to_fits(file_path=path.join(test_data_path, "array.fits"))
+    from autoconf.fitsable import output_to_fits
+    output_to_fits(values=arr.native.array.astype("float"), file_path=path.join(test_data_path, "array.fits"), header_dict=arr.mask.header_dict)
 
     array_from_out = aa.Array1D.from_fits(
         file_path=path.join(test_data_path, "array.fits"), hdu=0, pixel_scales=1.0

test_autoarray/structures/arrays/test_uniform_2d.py

@@ -213,7 +213,8 @@ def test__output_to_fits__3x3_ones__fits_file_has_ones_and_correct_pixel_scale_h
 
     os.makedirs(test_data_path)
 
-    array_2d.output_to_fits(file_path=path.join(test_data_path, "array.fits"))
+    from autoconf.fitsable import output_to_fits
+    output_to_fits(values=array_2d.native.array.astype("float"), file_path=path.join(test_data_path, "array.fits"), header_dict=array_2d.mask.header_dict)
 
     array_from_fits = aa.Array2D.from_fits(
         file_path=path.join(test_data_path, "array.fits"), hdu=0, pixel_scales=1.0

test_autoarray/structures/grids/test_uniform_2d.py

@@ -475,7 +475,8 @@ def test__to_and_from_fits_methods():
 
     file_path = path.join(test_grid_dir, "grid_2d.fits")
 
-    grid_2d.output_to_fits(file_path=file_path, overwrite=True)
+    from autoconf.fitsable import output_to_fits
+    output_to_fits(values=grid_2d.native.array.astype("float"), file_path=file_path, overwrite=True, header_dict=grid_2d.mask.header_dict)
 
     grid_from_fits = aa.Grid2D.from_fits(file_path=file_path, pixel_scales=2.0)
 

test_autoarray/structures/test_visibilities.py

@@ -102,7 +102,8 @@ def test__output_to_fits():
 
     os.makedirs(test_data_path)
 
-    visibilities.output_to_fits(file_path=path.join(test_data_path, "data.fits"))
+    from autoconf.fitsable import output_to_fits
+    output_to_fits(values=visibilities.in_array, file_path=path.join(test_data_path, "data.fits"))
 
     visibilities_from_out = aa.Visibilities.from_fits(
         file_path=path.join(test_data_path, "data.fits"), hdu=0