def test_spectrum1dref_copy():
"""
Test the ability to duplicate a Spectrum1DRef object.
"""
wcs = WCS({"CUNIT1": "Angstrom", "CRVAL1": 200.0, "CDELT1": 1.0})
data = np.random.sample(100)
disp = np.arange(100)
spectrum = Spectrum1DRef.from_array(data,
dispersion=disp,
unit=u.Unit("Jy"),
dispersion_unit=u.Unit("Angstrom"),
wcs=wcs,
meta={'author': 'me'})
spectrum_copy = Spectrum1DRef.copy(spectrum)
# Check data
assert (spectrum.data == spectrum_copy.data).all()
assert (spectrum.dispersion == spectrum_copy.dispersion).all()
# Check unit, wcs, meta
assert spectrum.unit == spectrum_copy.unit
assert spectrum.wcs.wcs.cunit[0] == spectrum_copy.wcs.wcs.cunit[0]
assert spectrum.wcs.wcs.crval[0] == spectrum_copy.wcs.wcs.crval[0]
assert spectrum.wcs.wcs.cdelt[0] == spectrum_copy.wcs.wcs.cdelt[0]
assert spectrum.meta == spectrum_copy.meta
def from_self(self, name="", layer_mask=None):
"""
Create a new, parentless, layer based on this layer
Parameters
----------
name: str
Name of the new layer
layer_mask: layer
The layer defining the valid data mask.
Returns
-------
new_layer:
The new, parentless, layer.
"""
gen_spec = Spectrum1DRef.copy(self, name=name)
return self.from_parent(parent=gen_spec,
layer_mask=layer_mask,
name=name)
def test_spectrum1dref_copy():
"""
Test the ability to duplicate a Spectrum1DRef object.
"""
wcs = WCS({"CUNIT1": "Angstrom", "CRVAL1": 200.0, "CDELT1": 1.0})
data = np.random.sample(100)
disp = np.arange(100)
spectrum = Spectrum1DRef.from_array(data, dispersion=disp,
unit=u.Unit("Jy"),
dispersion_unit=u.Unit("Angstrom"),
wcs=wcs, meta={'author': 'me'})
spectrum_copy = Spectrum1DRef.copy(spectrum)
# Check data
assert (spectrum.data == spectrum_copy.data).all()
assert (spectrum.dispersion == spectrum_copy.dispersion).all()
# Check unit, wcs, meta
assert spectrum.unit == spectrum_copy.unit
assert spectrum.wcs.wcs.cunit[0] == spectrum_copy.wcs.wcs.cunit[0]
assert spectrum.wcs.wcs.crval[0] == spectrum_copy.wcs.wcs.crval[0]
assert spectrum.wcs.wcs.cdelt[0] == spectrum_copy.wcs.wcs.cdelt[0]
assert spectrum.meta == spectrum_copy.meta
def from_self(self, name="", layer_mask=None):
gen_spec = Spectrum1DRef.copy(self, name=name)
return self.from_parent(parent=gen_spec, layer_mask=layer_mask)
def from_self(self, name="", layer_mask=None):
gen_spec = Spectrum1DRef.copy(self, name=name)
return self.from_parent(parent=gen_spec, layer_mask=layer_mask)
