Commit e50e6308 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add test_prepare_fits_files_for_ingestion()

parent e5a99a1c

tests/test_fits.py

+6 −0
Original line number Diff line number Diff line
@@ -52,3 +52,9 @@ class TestFitsHeaderOps(unittest.TestCase): 
        hl = fits.open(test_fits_file)

        fits_bytes = hl.to_bytes()

        self.assertIsInstance(fits_bytes, bytes)



    def test_prepare_fits_files_for_ingestion(self):

        ingestion_files = fits.prepare_fits_files_for_ingestion([test_fits_file])

        self.assertIsInstance(ingestion_files, list)

        self.assertEqual(len(ingestion_files), 1)

        self.assertEqual(set(ingestion_files[0].keys()), {"file_name", "file_content"})