def _generateFakeSky(self):
skySim = SkySim()
skySim.addStarByRaDecInDeg(0, 1.0, 1.0, 17.0)
starInstSettingFile = os.path.join(getModulePath(), "configData",
"instFile", "starDefault.inst")
return skySim, starInstSettingFile
def _prepareSkySim(opdMetr, starMag):
skySim = SkySim()
starId = 0
raInDegList, declInDegList = opdMetr.getFieldXY()
for raInDeg, declInDeg in zip(raInDegList, declInDegList):
# It is noted that the field position might be < 0. But it is not the
# same case for ra (0 <= ra <= 360).
if (raInDeg < 0):
raInDeg += 360.0
skySim.addStarByRaDecInDeg(starId, raInDeg, declInDeg, starMag)
starId += 1
return skySim
class TestSkySim(unittest.TestCase):
""" Test the SkySim class."""
def setUp(self):
# Directory to the focal plane file
folderPath2FocalPlane = os.path.join(getModulePath(), "tests",
"testData", "testOpdFunc")
self.skySim = SkySim()
self.skySim.setFolderPath2FocalPlane(folderPath2FocalPlane)
def testAddStarByRaDecInDeg(self):
self.skySim.addStarByRaDecInDeg(1, 2, 3, 4)
self.assertEqual(len(self.skySim.starId), 1)
self.skySim.addStarByRaDecInDeg(2, 2.1, 3, 4)
self.assertEqual(len(self.skySim.starId), 2)
self.assertEqual(self.skySim.starId[0], 1)
self.assertEqual(self.skySim.starId[1], 2)
# Try to add the same star Id again
self.skySim.addStarByRaDecInDeg(2, 2.1, 3, 4)
self.assertEqual(len(self.skySim.starId), 2)
def testResetSky(self):
self.skySim.addStarByRaDecInDeg(1, 2, 3, 4)
self.skySim.resetSky()
self.assertEqual(len(self.skySim.starId), 0)
def testSetStarRaDecInDeg(self):
self.skySim.setStarRaDecInDeg(np.array([0]), np.array([1]),
np.array([2]), np.array([3]))
self.assertEqual(len(self.skySim.starId), 1)
def testAddStarByFile(self):
self._addStarByFile()
self.assertEqual(len(self.skySim.starId), 8)
self.assertEqual(self.skySim.ra[2], -1.176)
self.assertEqual(self.skySim.decl[2], 1.196)
self.assertEqual(self.skySim.mag[2], 17.0)
def _addStarByFile(self):
skyFile = os.path.join(getModulePath(), "tests", "testData", "sky",
"wfsStar.txt")
self.skySim.addStarByFile(skyFile)
def testExportSkyToFile(self):
self._addStarByFile()
outputFilePath = os.path.join(getModulePath(), "output",
"testSkyOutput.txt")
self.skySim.exportSkyToFile(outputFilePath)
self.assertTrue(os.path.isfile(outputFilePath))
os.remove(outputFilePath)
def testAddStarByChipPos(self):
self._setObservationMetaData()
# Add the star
sensorName = "R22_S11"
starId = 0
xInpixelInCam = 2000
yInPixelInCam = 2036
starMag = 17
self.skySim.addStarByChipPos(sensorName, starId, xInpixelInCam,
yInPixelInCam, starMag)
# Test the result
self.assertAlmostEqual(self.skySim.ra[0], 359.99971038)
self.assertAlmostEqual(self.skySim.decl[0], 0.0001889)
def _setObservationMetaData(self):
ra = 0
decl = 0
rotSkyPos = 0
mjd = 59580.0
self.skySim.setObservationMetaData(ra, decl, rotSkyPos, mjd)
def _addSglStarToSkySim(self):
skySim = SkySim()
skySim.addStarByRaDecInDeg(0, 0.1, 0.2, 5.0)
return skySim
class TestSkySim(unittest.TestCase):
""" Test the SkySim class."""
def setUp(self):
self.skySim = SkySim()
def testGetStarId(self):
starId = self.skySim.getStarId()
self.assertEqual(len(starId), 0)
self.assertTrue(isinstance(starId, np.ndarray))
def testGetRaDecInDeg(self):
ra, decl = self.skySim.getRaDecInDeg()
self.assertEqual(len(ra), 0)
self.assertTrue(isinstance(ra, np.ndarray))
self.assertEqual(len(decl), 0)
self.assertTrue(isinstance(decl, np.ndarray))
def testGetStarMag(self):
mag = self.skySim.getStarMag()
self.assertEqual(len(mag), 0)
self.assertTrue(isinstance(mag, np.ndarray))
def testAddStarByRaDecInDeg(self):
self.skySim.addStarByRaDecInDeg(1, 2, 3, 4)
self.assertEqual(len(self.skySim.getStarId()), 1)
self.skySim.addStarByRaDecInDeg(2, 2.1, 3, 4)
starId = self.skySim.getStarId()
self.assertEqual(len(starId), 2)
self.assertEqual(starId[0], 1)
self.assertEqual(starId[1], 2)
# Try to add the same star Id again
self.skySim.addStarByRaDecInDeg(2, 2.1, 3, 4)
self.assertEqual(len(self.skySim.getStarId()), 2)
def testResetSky(self):
self.skySim.addStarByRaDecInDeg(1, 2, 3, 4)
self.skySim.resetSky()
self.assertEqual(len(self.skySim.getStarId()), 0)
def testSetStarRaDecInDeg(self):
self.skySim.setStarRaDecInDeg(np.array([0]), np.array([1]),
np.array([2]), np.array([3]))
self.assertEqual(len(self.skySim.getStarId()), 1)
def testAddStarByFile(self):
self._addStarByFile("wfsStar.txt")
self.assertEqual(len(self.skySim.getStarId()), 8)
ra, decl = self.skySim.getRaDecInDeg()
self.assertEqual(ra[2], -1.176)
self.assertEqual(decl[2], 1.196)
self.assertEqual(self.skySim.getStarMag()[2], 17.0)
def _addStarByFile(self, skyFileName):
skyFile = os.path.join(getModulePath(), "tests", "testData", "sky",
skyFileName)
self.skySim.addStarByFile(skyFile)
def testAddStarByFileWithSglStar(self):
self._addStarByFile("wfsSglStar.txt")
self.assertEqual(len(self.skySim.getStarId()), 1)
ra, decl = self.skySim.getRaDecInDeg()
self.assertEqual(ra[0], 1.196)
self.assertEqual(decl[0], 1.176)
self.assertEqual(self.skySim.getStarMag()[0], 17.0)
def testExportSkyToFile(self):
self._addStarByFile("wfsStar.txt")
outputFilePath = os.path.join(getModulePath(), "output",
"testSkyOutput.txt")
self.skySim.exportSkyToFile(outputFilePath)
self.assertTrue(os.path.isfile(outputFilePath))
os.remove(outputFilePath)
def testAddStarByChipPos(self):
self._setObservationMetaData()
# Add the star
sensorName = "R22_S11"
starId = 0
xInpixelInCam = 2000
yInPixelInCam = 2036
starMag = 17
self.skySim.addStarByChipPos(sensorName, starId, xInpixelInCam,
yInPixelInCam, starMag)
# Test the result
ra, decl = self.skySim.getRaDecInDeg()
self.assertAlmostEqual(ra[0], 359.99971038)
self.assertAlmostEqual(decl[0], 0.0001889)
def _setObservationMetaData(self):
ra = 0
decl = 0
rotSkyPos = 0
mjd = 59580.0
self.skySim.setObservationMetaData(ra, decl, rotSkyPos, mjd)