Use PhotometryData in tests

mwcraig · Nov 22, 2023 · 2608c52 · 2608c52
1 parent 1b5c6e8
commit 2608c52
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/stellarphot/differential_photometry/aij_rel_fluxes.py b/stellarphot/differential_photometry/aij_rel_fluxes.py
@@ -6,6 +6,7 @@
 
 __all__ = ['add_in_quadrature', 'calc_aij_relative_flux']
 
+
 def add_in_quadrature(array):
     """
     Add an array of numbers in quadrature.
@@ -23,8 +24,8 @@ def calc_aij_relative_flux(star_data, comp_stars,
     Parameters
     ----------
 
-    star_data : '~astropy.table.Table'
-        Table of star data from one or more images.
+    star_data : 'stellarphot.PhotometryDat'
+        Photometry data from one or more images.
 
     comp_stars : '~astropy.table.Table'
         Table of comparison stars in the field. Must contain a column
@@ -56,7 +57,7 @@ def calc_aij_relative_flux(star_data, comp_stars,
     Returns
     -------
 
-    `astropy.table.Table` or None
+    `stellarphot.PhotometryData` or None
         The return type depends on the value of ``in_place``. If it is
         ``False``, then the new columns are returned as a separate table,
         otherwise the columns are simply added to the input table.

diff --git a/stellarphot/differential_photometry/tests/test_aij_rel_fluxes.py b/stellarphot/differential_photometry/tests/test_aij_rel_fluxes.py
@@ -7,6 +7,7 @@
 from astropy.time import Time
 import astropy.units as u
 
+from stellarphot import PhotometryData
 from stellarphot.differential_photometry.aij_rel_fluxes import calc_aij_relative_flux
 
 
@@ -54,6 +55,7 @@ def _raw_photometry_table():
                       names=['date-obs', 'ra', 'dec', 'aperture_net_cnts',
                              'noise_electrons', 'star_id'])
 
+    photom = PhotometryData(raw_table)
     return expected_flux_ratios, expected_flux_error, raw_table, raw_table[1:4]