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)