def test_image_store(self):
fits_object = fits.open(FITS_FILE)
expected_data = cPickle.dumps(fits_object[0].data)
expected_header = fits_object[0].header
store_fits([self.image], [expected_data], [str(expected_header)])
fetched_image = self.db.session.query(Image).filter(Image.id==self.image.id).first()
returned_data = cPickle.loads(fetched_image.data.fits_data)
returned_header = Header.fromstring(fetched_image.data.fits_header)
self.assertTrue((returned_data, expected_data))
self.assertEqual(returned_header, expected_header)
def test_image_store(self):
fits_object = fits.open(FITS_FILE)
expected_data = cPickle.dumps(fits_object[0].data)
expected_header = fits_object[0].header
store_fits([self.image], [expected_data], [str(expected_header)])
fetched_image = self.db.session.query(Image).filter(
Image.id == self.image.id).first()
returned_data = cPickle.loads(fetched_image.data.fits_data)
returned_header = Header.fromstring(fetched_image.data.fits_header)
self.assertTrue((returned_data, expected_data))
self.assertEqual(returned_header, expected_header)
def store_image_data(db_images, fits_datas, fits_headers):
logger.info("storing {} images to database".format(len(db_images)))
store_fits(db_images, fits_datas, fits_headers)
def store_image_data(db_images, fits_datas, fits_headers):
logger.info("storing {} images to database".format(len(db_images)))
store_fits(db_images, fits_datas, fits_headers)
