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

add support for fits: write test FITS file in setUp

parent c3da557f
Pipeline #10808 failed with stage
in 0 seconds
...@@ -12,6 +12,7 @@ Modified-History: ...@@ -12,6 +12,7 @@ Modified-History:
import os import os
import unittest import unittest
import numpy as np
from csst_common import fits from csst_common import fits
test_fits_file = os.path.join( test_fits_file = os.path.join(
...@@ -22,6 +23,15 @@ test_fits_file = os.path.join( ...@@ -22,6 +23,15 @@ test_fits_file = os.path.join(
class TestFitsHeaderOps(unittest.TestCase): class TestFitsHeaderOps(unittest.TestCase):
def setUp(self):
hl = fits.HDUList(
[
fits.PrimaryHDU(),
fits.ImageHDU(data=np.random.randn(5, 5)),
]
)
hl.writeto(test_fits_file)
def test_fits_open(self): def test_fits_open(self):
hl = fits.open(test_fits_file) hl = fits.open(test_fits_file)
self.assertIsInstance(hl, fits.HDUList) self.assertIsInstance(hl, fits.HDUList)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment