def testArrayValues(self):
"""Check that only the final value is used from an array
"""
metadata = PropertyList()
# work around DM-13232 by setting ABOOL one value at a time
for value in [True, True, True, False]:
metadata.add("ABOOL", value)
dataList = [
("AFLOAT", [1.2e25, -5.6]),
("ANINT", [-5, 752, 1052]),
("ASTRING1", ["value for string", "more"]),
]
for name, value in dataList:
metadata.set(name, value)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"ABOOL = F",
"AFLOAT = -5.6",
"ANINT = 1052",
"ASTRING1= 'more'",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertHeadersEqual(header, expectedHeader)
self.checkExcludeNames(metadata, expectedLines)
def testAgainstAstropyWcs(self):
skyWcs = makeSkyWcs(self.metadata, strip=False)
header = makeLimitedFitsHeader(self.metadata)
astropyWcs = astropy.wcs.WCS(header)
self.assertSkyWcsAstropyWcsAlmostEqual(skyWcs=skyWcs,
astropyWcs=astropyWcs,
bbox=self.bbox)
def testArrayValues(self):
"""Check that only the final value is used from an array
"""
metadata = PropertyList()
# work around DM-13232 by setting ABOOL one value at a time
for value in [True, True, True, False]:
metadata.add("ABOOL", value)
dataList = [
("AFLOAT", [1.2e25, -5.6]),
("ANINT", [-5, 752, 1052]),
("ASTRING1", ["value for string", "more"]),
]
for name, value in dataList:
metadata.set(name, value)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"ABOOL = F",
"AFLOAT = -5.6",
"ANINT = 1052",
"ASTRING1= 'more'",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertHeadersEqual(header, expectedHeader)
self.checkExcludeNames(metadata, expectedLines)
def testBasics(self):
"""Check basic formatting and skipping bad values
"""
metadata = PropertyList()
dataList = [
("ABOOL", True),
("AFLOAT", 1.2e25),
("ANINT", -5),
("LONGNAME1", 1), # name is longer than 8 characters; skip it
("LONGSTR", "skip this item because the formatted value "
"is too long: longer than 80 characters "),
("ASTRING1", "value for string"),
]
for name, value in dataList:
metadata.set(name, value)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"ABOOL = 1",
"AFLOAT = 1.2E+25",
"ANINT = -5",
"ASTRING1= 'value for string'",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertEqual(header, expectedHeader)
self.checkExcludeNames(metadata, expectedLines)
def testAgainstAstropyWcs(self):
skyWcs = makeSkyWcs(self.metadata, strip=False)
header = makeLimitedFitsHeader(self.metadata)
astropyWcs = astropy.wcs.WCS(header)
bbox = Box2D(Point2D(-1000, -1000), Extent2D(3000, 3000))
self.assertSkyWcsAstropyWcsAlmostEqual(skyWcs=skyWcs,
astropyWcs=astropyWcs,
bbox=bbox)
def checkExcludeNames(self, metadata, expectedLines):
"""Check that makeLimitedFitsHeader properly excludes specified names
"""
names = metadata.names()
# skip each name in turn, then skip all names at once
excludeNamesList = [set([name]) for name in names]
excludeNamesList.append(set(names))
for excludeNames in excludeNamesList:
header = makeLimitedFitsHeader(metadata, excludeNames=excludeNames)
expectedHeader = "".join("%-80s" % val for val in expectedLines
if val[0:8].strip() not in excludeNames)
self.assertHeadersEqual(header, expectedHeader)
def checkExcludeNames(self, metadata, expectedLines):
"""Check that makeLimitedFitsHeader properly excludes specified names
"""
names = metadata.names()
# skip each name in turn, then skip all names at once
excludeNamesList = [set([name]) for name in names]
excludeNamesList.append(set(names))
for excludeNames in excludeNamesList:
header = makeLimitedFitsHeader(metadata, excludeNames=excludeNames)
expectedHeader = "".join("%-80s" % val for val in expectedLines
if val[0:8].strip() not in excludeNames)
self.assertHeadersEqual(header, expectedHeader)
def testBasics(self):
"""Check basic formatting and skipping bad values
"""
metadata = PropertyList()
dataList = [
("ABOOL", True),
("AFLOAT", 1.2e25),
("AFLOAT2", 1.0e30),
("ANINT", -5),
("AFLOATZ", 0.0), # ensure a float stays a float
("INTFLOAT", -5.0),
("LONGFLT", 0.0089626337538440005),
("ANUNDEF", None),
("LONGNAME1", 1), # name is longer than 8 characters; skip it
("LONGSTR", "skip this item because the formatted value "
"is too long: longer than 80 characters "),
("ASTRING1", "value for string"),
("ANAN", float("NaN"))
]
for name, value in dataList:
metadata.set(name, value)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"ABOOL = T",
"AFLOAT = 1.2E+25",
"AFLOAT2 = 1E+30",
"ANINT = -5",
"AFLOATZ = 0.0",
"INTFLOAT= -5.0",
"LONGFLT = 0.0089626337538440005",
"ANUNDEF =",
"ASTRING1= 'value for string'",
"ANAN =",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertHeadersEqual(header, expectedHeader)
self.checkExcludeNames(metadata, expectedLines)
def testSinglePrecision(self):
"""Check that single precision floats do work"""
metadata = PropertyList()
# Numeric form of single precision floats need smaller precision
metadata.setFloat("SINGLE", 3.14159)
metadata.setFloat("SINGLEI", 5.0)
metadata.setFloat("SINGLEE", -5.9e20)
metadata.setFloat("EXP", -5e10)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"SINGLE = 3.14159",
"SINGLEI = 5.0",
"SINGLEE = -5.9E+20",
"EXP = -5E+10",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertHeadersEqual(header, expectedHeader, rtol=np.finfo(np.float32).resolution)
def testSinglePrecision(self):
"""Check that single precision floats do work"""
metadata = PropertyList()
# Numeric form of single precision floats need smaller precision
metadata.setFloat("SINGLE", 3.14159)
metadata.setFloat("SINGLEI", 5.0)
metadata.setFloat("SINGLEE", -5.9e20)
metadata.setFloat("EXP", -5e10)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"SINGLE = 3.14159",
"SINGLEI = 5.0",
"SINGLEE = -5.9E+20",
"EXP = -5E+10",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertHeadersEqual(header, expectedHeader, rtol=np.finfo(np.float32).resolution)
def testBasics(self):
"""Check basic formatting and skipping bad values
"""
metadata = PropertyList()
dataList = [
("ABOOL", True),
("AFLOAT", 1.2e25),
("AFLOAT2", 1.0e30),
("ANINT", -5),
("AFLOATZ", 0.0), # ensure a float stays a float
("INTFLOAT", -5.0),
("LONGFLT", 0.0089626337538440005),
("ANUNDEF", None),
("LONGNAME1", 1), # name is longer than 8 characters; skip it
("LONGSTR", "skip this item because the formatted value "
"is too long: longer than 80 characters "),
("ASTRING1", "value for string"),
]
for name, value in dataList:
metadata.set(name, value)
header = makeLimitedFitsHeader(metadata)
expectedLines = [ # without padding to 80 chars
"ABOOL = T",
"AFLOAT = 1.2E+25",
"AFLOAT2 = 1E+30",
"ANINT = -5",
"AFLOATZ = 0.0",
"INTFLOAT= -5.0",
"LONGFLT = 0.0089626337538440005",
"ANUNDEF =",
"ASTRING1= 'value for string'",
]
expectedHeader = "".join("%-80s" % val for val in expectedLines)
self.assertHeadersEqual(header, expectedHeader)
self.checkExcludeNames(metadata, expectedLines)
def testAgainstAstropyWcs(self):
bbox = Box2D(Point2D(-1000, -1000), Extent2D(2000, 2000))
for crval, orientation, flipX, projection in itertools.product(
self.crvalList, self.orientationList, (False, True),
("TAN", "STG", "CEA", "AIT")):
cdMatrix = makeCdMatrix(scale=self.scale,
orientation=orientation,
flipX=flipX)
metadata = makeSimpleWcsMetadata(crpix=self.crpix,
crval=crval,
cdMatrix=cdMatrix,
projection=projection)
header = makeLimitedFitsHeader(metadata)
astropyWcs = astropy.wcs.WCS(header)
skyWcs = makeSkyWcs(crpix=self.crpix,
crval=crval,
cdMatrix=cdMatrix,
projection=projection)
# Most projections only seem to agree to within 1e-4 in the round trip test
self.assertSkyWcsAstropyWcsAlmostEqual(skyWcs=skyWcs,
astropyWcs=astropyWcs,
bbox=bbox)