def testStoreHeaders(self):
"""Tests whether store header executes on an SDSS frame; this verifies
the created standardized dictionary has correcly named keys.
"""
data = MockTmpUploadedFile("reduced_frame-i-008108-5-0025.fits",
self.testDataDir)
fits = TemporaryUploadedFileWrapper(data)
fitsProcessor = UploadProcessor.fromFileWrapper(fits)
fitsProcessor.standardizeHeader()
def testStoreThumbnails(self):
"""Tests whether two thumbnails appear at the expected location."""
data = MockTmpUploadedFile("reduced_frame-i-008108-5-0025.fits",
self.testDataDir)
fits = TemporaryUploadedFileWrapper(data)
fitsProcessor = UploadProcessor.fromFileWrapper(fits)
fitsProcessor.createThumbnails()
large = os.path.join(self.tmpTestDir, fits.basename + '_large.jpg')
small = os.path.join(self.tmpTestDir, fits.basename + '_small.jpg')
self.assertTrue(os.path.exists(large))
self.assertTrue(os.path.exists(small))
def setUp(self):
self.tmpTestDir = tempfile.mkdtemp(dir=TESTDIR)
TemporaryUploadedFileWrapper.save_root = self.tmpTestDir
UploadProcessor.media_root = self.tmpTestDir
fnames = os.listdir(self.testDataDir)
self.fits = []
for fname in fnames:
if "fits" in fname or "fit" in fname:
tmp = MockTmpUploadedFile(fname, self.testDataDir)
self.fits.append(TemporaryUploadedFileWrapper(tmp))
with open(
os.path.join(self.testDataDir,
"expectedStandardizedValues.yaml")) as f:
self.standardizedAnswers = yaml.safe_load(f)
def setUp(self):
self.tmpTestDir = tempfile.mkdtemp(dir=TESTDIR)
TemporaryUploadedFileWrapper.save_root = self.tmpTestDir
Thumbnails.SMALL_THUMB_ROOT = self.tmpTestDir
Thumbnails.LARGE_THUMB_ROOT = self.tmpTestDir
header_standardizer.ASTROMETRY_KEY = "test"
header_standardizer.ASTRONET_CLIENT = MockAstrometryServer()
fnames = os.listdir(self.testDataDir)
self.fits = []
for fname in fnames:
if "fits" in fname or "fit" in fname:
tmp = MockTmpUploadedFile(fname, self.testDataDir)
self.fits.append(TemporaryUploadedFileWrapper(tmp))
with open(
os.path.join(self.testDataDir,
"expectedStandardizedValues.yaml")) as f:
self.standardizedAnswers = yaml.safe_load(f)
def setUp(self):
self.tmpTestDir = tempfile.mkdtemp(dir=TESTDIR)
TemporaryUploadedFileWrapper.save_root = self.tmpTestDir
self.testData = {
# filename: (basename, ext)
"reduced_A3671-C2018_F4-R-3.fit":
("reduced_A3671-C2018_F4-R-3", ".fit"),
"reduced_bi327715.fits": ("reduced_bi327715", ".fits"),
"reduced_c4d_200306_000415_ori.fits.fz":
("reduced_c4d_200306_000415_ori", ".fits.fz"),
"reduced_calexp-0941420_23.fits":
("reduced_calexp-0941420_23", ".fits"),
"reduced_HSCA21787010.fits": ("reduced_HSCA21787010", ".fits"),
"reduced_frame-i-008108-5-0025.fits":
("reduced_frame-i-008108-5-0025", ".fits"),
}
self.fits = []
for fname in self.testData.keys():
tmp = MockTmpUploadedFile(fname, self.testDataDir)
self.fits.append(TemporaryUploadedFileWrapper(tmp))