Coverage for tests/test_WavelengthArray.py : 27%

Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
"""Test the basic behaviour of WavelenthArray""" minWl = 600 maxWl = 900 size = 50
# Creation wlArray = WavelengthArray(minWl, maxWl, size) linspace = np.linspace(minWl, maxWl, size, dtype=np.float32) self.assertEqual(len(wlArray), size) self.assertEqual(len(wlArray), len(linspace)) self.assertFloatsEqual(wlArray, linspace)
# Iteration for xx, yy in zip(wlArray, linspace): self.assertFloatsEqual(xx, yy)
# Indexing for ii in range(size): self.assertFloatsEqual(wlArray[ii], linspace[ii]) self.assertFloatsEqual(wlArray[-ii], linspace[-ii])
# Behaves like a numpy array self.assertFloatsEqual(wlArray.mean(), linspace.mean()) self.assertFloatsEqual(wlArray.std(), linspace.std())
"""Test that we can round-trip a WavelengthArray""" size = 50 wlArray = WavelengthArray(600, 900, size) header = wlArray.toFitsHeader() copy = WavelengthArray.fromFitsHeader(header, size) self.assertEqual(len(copy), size) self.assertFloatsEqual(copy, wlArray) self.assertEqual(type(copy), type(wlArray))
"""Test that the WavelengthArray WCS header is correct""" size = 50 wlArray = WavelengthArray(600, 900, size) header = wlArray.toFitsHeader() wcs = astropy.wcs.WCS(header) for ii in range(size): self.assertFloatsAlmostEqual(wlArray[ii], wcs.pixel_to_world(ii + 1).to(astropy.units.nm).value, atol=1.0e-4)
lsst.utils.tests.init()
setup_module(sys.modules["__main__"]) from argparse import ArgumentParser parser = ArgumentParser(__file__) parser.add_argument("--display", help="Display backend") args, argv = parser.parse_known_args() display = args.display unittest.main(failfast=True, argv=[__file__] + argv) |