append unittest:dark

tests/test_darknoise_func.py

+import unittest
+
+import sys,os,math
+from itertools import islice
+import numpy as np
+import galsim
+import yaml
+
+from ObservationSim.Instrument import Chip, Filter, FilterParam, FocalPlane
+#from ObservationSim.Instrument.Chip import ChipUtils as chip_utils
+
+### test FUNCTION --- START ###
+def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150., InputDark=None):
+    if InputDark == None:
+        # base_level = read_noise**2 + dark_noise*(exptime+0.5*readout_time)
+        ## base_level = dark_noise*(exptime+0.5*readout_time)
+        base_level = dark_noise*(exptime)
+        base_img1 = base_level * np.ones_like(img.array)
+    else:
+        base_img1 = np.zeros_like(img.array)
+
+    ny = int(chip.npix_y/2)
+    nx = chip.npix_x
+    arr = np.arange(ny).reshape(ny, 1)
+    arr = np.broadcast_to(arr, (ny, nx))
+    base_img2 = np.zeros_like(img.array)
+    base_img2[:ny, :] = arr
+    base_img2[ny:, :] = arr[::-1,:]
+    base_img2[:,:]    = base_img2[:,:]*(readout_time/ny)*dark_noise
+    return base_img1+base_img2
+### test FUNCTION --- END ###
+
+
+def defineCCD(iccd, config_file):
+    with open(config_file, "r") as stream:
+        try:
+            config = yaml.safe_load(stream)
+            #for key, value in config.items():
+            #    print (key + " : " + str(value))
+        except yaml.YAMLError as exc:
+            print(exc)
+    chip = Chip(chipID=iccd, config=config)
+    chip.img = galsim.ImageF(chip.npix_x, chip.npix_y)
+    focal_plane = FocalPlane(chip_list=[iccd])
+    chip.img.wcs= focal_plane.getTanWCS(192.8595, 27.1283, -113.4333*galsim.degrees, chip.pix_scale)
+    return chip
+
+def defineFilt(chip):
+    filter_param = FilterParam()
+    filter_id, filter_type = chip.getChipFilter()
+    filt = Filter(
+        filter_id=filter_id,
+        filter_type=filter_type,
+        filter_param=filter_param,
+        ccd_bandpass=chip.effCurve)
+    bandpass_list = filt.bandpass_sub_list
+    return bandpass_list
+
+
+class detModule_coverage(unittest.TestCase):
+    def __init__(self, methodName='runTest'):
+        super(detModule_coverage, self).__init__(methodName)
+        self.dataPath = "/public/home/chengliang/CSSOSDataProductsSims/csst-simulation/tests/UNIT_TEST_DATA" ##os.path.join(os.getenv('UNIT_TEST_DATA_ROOT'), 'csst_fz_gc1')
+        self.iccd = 1
+
+    def test_add_dark(self):
+        config_file = os.path.join(self.dataPath, 'config_test.yaml')
+        chip = defineCCD(self.iccd, config_file)
+        bandpass = defineFilt(chip)
+        print(chip.chipID)
+        print(chip.cen_pix_x, chip.cen_pix_y)
+
+        exptime=150.
+        base_img = get_base_img(img=chip.img, chip=chip, read_noise=chip.read_noise, readout_time=chip.readout_time, dark_noise=chip.dark_noise, exptime=exptime, InputDark=None)
+
+        ny = int(chip.npix_y/2)
+        self.assertTrue( np.abs(np.max(base_img) - (exptime*chip.dark_noise+(ny-1)*(chip.readout_time/ny)*chip.dark_noise )) < 1e-6 )
+        self.assertTrue( np.min(base_img) == 3 )
+
+        base_img = get_base_img(img=chip.img, chip=chip, read_noise=chip.read_noise, readout_time=chip.readout_time, dark_noise=chip.dark_noise, exptime=150., InputDark="testTag")
+        self.assertTrue( np.abs(np.max(base_img) - ((ny-1)*(chip.readout_time/ny)*chip.dark_noise )) < 1e-6 )
+        
+
+
+if __name__ == '__main__':
+    unittest.main()