def setUp(self):
import pkg_resources
import tempfile
file_name = 'data/sample_data_NGC7023.tbl'
file_path = pkg_resources.resource_filename('pypahdb', file_name)
self.observation = Observation(file_path)
self.decomposer = Decomposer(self.observation.spectrum)
self.tmpdir = tempfile.gettempdir()
class DecomposerTestCase(unittest.TestCase):
"""Unit tests for `decomposer.py`"""
def setUp(self):
import pkg_resources
import tempfile
file_name = 'data/sample_data_NGC7023.tbl'
file_path = pkg_resources.resource_filename('pypahdb', file_name)
self.observation = Observation(file_path)
self.decomposer = Decomposer(self.observation.spectrum)
self.tmpdir = tempfile.gettempdir()
def test_is_instance(self):
"""Can we create an instance of Decomposer?"""
self.assertIsInstance(self.decomposer, Decomposer)
def test_has_fit(self):
"""Can we create a fit?"""
self.assertIsInstance(self.decomposer.fit, np.ndarray)
def test_has_ionized_fraction(self):
"""Can we create an ionized fraction?"""
self.assertIsInstance(self.decomposer.ionized_fraction, np.ndarray)
def test_has_large_fraction(self):
"""Can we calculate a large fraction?"""
self.assertIsInstance(self.decomposer.large_fraction, np.ndarray)
def test_has_charge(self):
"""Can we generate a charge results?"""
self.assertIsInstance(self.decomposer.charge, dict)
def test_has_size(self):
"""Can we generate a size results?"""
self.assertIsInstance(self.decomposer.size, dict)
def test_do_pdf(self):
"""Can we output a PDF?"""
ofile = os.path.join(self.tmpdir, "result.pdf")
self.decomposer.save_pdf(ofile)
self.assertTrue(os.path.isfile(ofile))
def test_do_fits(self):
"""Can we output FITS?"""
ofile = os.path.join(self.tmpdir, "result.pdf")
self.decomposer.save_fits(ofile)
self.assertTrue(os.path.isfile(ofile))
#!/usr/bin/env python3
"""
example.py
Example of using pypahdb to decompose an astronomical PAH spectrum.
"""
import pkg_resources
from pypahdb.decomposer import Decomposer
from pypahdb.observation import Observation
if __name__ == '__main__':
# Sample data (in FITS format).
file_path = 'resources/sample_data_NGC7023.fits'
data_file = pkg_resources.resource_filename('pypahdb', file_path)
# Construct an Observation object.
obs = Observation(data_file)
# Pass the Observation's spectrum to Decomposer, which performs the fit.
pahdb_fit = Decomposer(obs.spectrum)
# Write the results to file.
# doplots is disabled due to significant CPU overhead
pahdb_fit.save_pdf('NGC7023_pypahdb_fits_example.pdf',
header=obs.header,
doplots=False)
pahdb_fit.save_fits('NGC7023_pypahdb_fits_example.fits', header=obs.header)