diff --git a/astrodbkit2/tests/test_spectra.py b/astrodbkit2/tests/test_spectra.py index f47d4e6..5a67a92 100644 --- a/astrodbkit2/tests/test_spectra.py +++ b/astrodbkit2/tests/test_spectra.py @@ -13,7 +13,8 @@ @pytest.fixture(scope="module") def good_spex_file(): - n = np.empty((564, 3)) + fake_data = np.arange(1, 10, 1) + n = np.array([fake_data, fake_data, fake_data]) hdr = fits.Header() hdr['TELESCOP'] = 'NASA IRTF' hdr['INSTRUME'] = 'SPeX, IRTF Spectrograph' @@ -26,7 +27,8 @@ def good_spex_file(): @pytest.fixture(scope="module") def bad_spex_file(): - n = np.empty((564, 3)) + fake_data = np.arange(1, 10, 1) + n = np.array([fake_data, fake_data, fake_data]) hdr = fits.Header() hdr['TELESCOP'] = 'MISSING' hdr['INSTRUME'] = 'MISSING'