Fix tests

Author: samaloney

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():

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():

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():

sunkit_spex/legacy/tests/test_thermal.py sunkit_spex/legacy/tests/test_thermal_.pysunkit_spex/legacy/tests/test_thermal.py renamed to 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(

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]))