From 358a6ce14b0813233c41e15e9b1426f7b07639a6 Mon Sep 17 00:00:00 2001 From: Ricky O'Steen Date: Tue, 15 Oct 2024 15:07:47 -0400 Subject: [PATCH] Better test improvement --- specutils/tests/test_region_extract.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/specutils/tests/test_region_extract.py b/specutils/tests/test_region_extract.py index 9ff5b3d92..2c46f6f12 100644 --- a/specutils/tests/test_region_extract.py +++ b/specutils/tests/test_region_extract.py @@ -111,10 +111,8 @@ def test_pixel_spectralaxis_extraction(): def test_slab_simple(simulated_spectra): - np.random.seed(42) - spectrum = simulated_spectra.s1_um_mJy_e1 - uncertainty = StdDevUncertainty(0.1*np.random.random(len(spectrum.flux))*u.mJy) + uncertainty = StdDevUncertainty(np.full(spectrum.flux.shape, 0.1) * u.mJy) spectrum.uncertainty = uncertainty sub_spectrum = spectral_slab(spectrum, 0.6*u.um, 0.8*u.um) @@ -220,7 +218,7 @@ def test_region_empty(): def test_region_descending(simulated_spectra): spectrum = simulated_spectra.s1_um_mJy_e1 - uncertainty = StdDevUncertainty(0.1*np.ones(len(spectrum.flux))*u.mJy) + uncertainty = StdDevUncertainty(np.full(spectrum.flux.shape, 0.1) * u.mJy) spectrum.uncertainty = uncertainty region = SpectralRegion(0.8*u.um, 0.6*u.um) @@ -250,7 +248,7 @@ def test_descending_spectral_axis(simulated_spectra): def test_region_two_sub(simulated_spectra): spectrum = simulated_spectra.s1_um_mJy_e1 - uncertainty = StdDevUncertainty(0.1*np.ones(len(spectrum.flux))*u.mJy) + uncertainty = StdDevUncertainty(np.full(spectrum.flux.shape, 0.1) * u.mJy) spectrum.uncertainty = uncertainty region = SpectralRegion([(0.6*u.um, 0.8*u.um), (0.86*u.um, 0.89*u.um)]) @@ -288,7 +286,7 @@ def test_region_two_sub(simulated_spectra): def test_bounding_region(simulated_spectra): spectrum = simulated_spectra.s1_um_mJy_e1 - uncertainty = StdDevUncertainty(0.1*np.ones(len(spectrum.flux))*u.mJy) + uncertainty = StdDevUncertainty(np.full(spectrum.flux.shape, 0.1) * u.mJy) spectrum.uncertainty = uncertainty region = SpectralRegion([(0.6*u.um, 0.8*u.um), (0.86*u.um, 0.89*u.um)])