Fix tests

sunpy · Jun 20, 2024 · fedd73c · fedd73c
1 parent ff60cef
commit fedd73c
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 5 deletions.
diff --git a/sunkit_spex/legacy/tests/test_brem.py b/sunkit_spex/legacy/tests/test_brem.py
@@ -2,8 +2,8 @@
 import pytest
 from numpy.testing import assert_allclose, assert_array_equal
 
-from sunkit_spex import emission
-from sunkit_spex.integrate import fixed_quad, gauss_legendre
+from sunkit_spex.legacy import emission
+from sunkit_spex.legacy.integrate import fixed_quad, gauss_legendre
 
 
 def test_broken_power_law_electron_distribution():

diff --git a/sunkit_spex/legacy/tests/test_integrate.py b/sunkit_spex/legacy/tests/test_integrate.py
@@ -1,7 +1,7 @@
 import numpy as np
 from numpy.testing import assert_allclose
 
-from sunkit_spex.integrate import fixed_quad, gauss_legendre
+from sunkit_spex.legacy.integrate import fixed_quad, gauss_legendre
 
 
 def test_scalar():

diff --git a/sunkit_spex/legacy/tests/test_photon_power_law.py b/sunkit_spex/legacy/tests/test_photon_power_law.py
@@ -3,7 +3,7 @@
 
 import astropy.units as u
 
-from sunkit_spex import photon_power_law as ppl
+from sunkit_spex.legacy import photon_power_law as ppl
 
 
 def test_different_bins():

diff --git a/sunkit_spex/legacy/tests/test_thermal.py → sunkit_spex/legacy/tests/test_thermal_.py b/sunkit_spex/legacy/tests/test_thermal.py → sunkit_spex/legacy/tests/test_thermal_.py
@@ -5,7 +5,7 @@
 
 import astropy.units as u
 
-from sunkit_spex import thermal
+from sunkit_spex.legacy import thermal
 
 # Manually load file that was used to compile expected flux values.
 thermal.setup_continuum_parameters(

diff --git a/sunkit_spex/spectrum/uncertainty.py b/sunkit_spex/spectrum/uncertainty.py
@@ -31,6 +31,7 @@ class PoissonUncertainty(_VariancePropagationMixin, NDUncertainty):
     `Poisson` should always be associated with an `NDData`-like
     instance, either by creating it during initialization::
 
+        >>> from astropy.nddata import NDData
         >>> from sunkit_spex.spectrum.uncertainty import PoissonUncertainty
         >>> ndd = NDData([1,2,3], unit='m',
         ...              uncertainty=PoissonUncertainty([0.1, 0.1, 0.1]))