Skip to content
GitLab
Commit 5174c719 authored by GZhao's avatar GZhao
Browse files

update pep8 for unittest code

parent 190f1129
Pipeline #7117 passed with stage
in 0 seconds
Hide whitespace changes
Inline Side-by-side
tests/test_camera.py
View file @ 5174c719
......@@ -49,6 +49,7 @@ class TestEMCCD(unittest.TestCase):
image = emccd.readout(image_focal, None, expt, iamge_cosmic_ray, emgain=emgain)
self.assertEqual(image.shape[0], emccd.bias_shape[0])
self.assertEqual(image.shape[1], emccd.bias_shape[1])
def test_em_fix_fun(self):
emccd = CpicVisEmccd()
emgain = emccd.em_fix_fuc_fit(-5)
......@@ -57,7 +58,7 @@ class TestEMCCD(unittest.TestCase):
def test_emccd_update(self):
emccd = CpicVisEmccd()
emccd.ccd_temp = -100
emgain = emccd.emgain_set(1024, ccd_temp=None, self_update=False )
emgain = emccd.emgain_set(1024, ccd_temp=None, self_update=False)
self.assertAlmostEqual(emgain, 1.23, places=2)
......@@ -74,5 +75,3 @@ class TestEMCCD(unittest.TestCase):
# bias_images = np.array(bias_images)
# from astropy.io import fits
# fits.writeto("bias.fits", bias_images)
tests/test_io.py
View file @ 5174c719
......@@ -9,7 +9,7 @@ import csst_cpic_sim.io as io
from astropy.io import fits
import numpy as np
import yaml
import os
import logging
cstar = {
'magnitude': 5,
......@@ -34,20 +34,19 @@ params = {
'EXPSTART': '2020-01-01T00:00:00.000',
'EXPEND': '2020-01-02T00:00:00.000',
'frame_info': [
{'expt_start': 0,
'expt_end': 20,
'platescale': 1.0,
'iwa': 0,
'chiptemp': 1}
{
'expt_start': 0,
'expt_end': 20,
'platescale': 1.0,
'iwa': 0,
'chiptemp': 1}
]
}
camera = CpicVisEmccd()
import logging
class TestIO(unittest.TestCase):
class TestIO(unittest.TestCase):
def test_obsid_parser(self):
self.assertRaises(ValueError, obsid_parser, '20190101')
self.assertRaises(ValueError, obsid_parser, '123456789012')
......@@ -99,7 +98,7 @@ rotation: 0
emgain: 100
emset: -1
skybg: 21
expstart:
expstart:
target:
cstar:
......@@ -125,5 +124,6 @@ target:
self.assertEqual(
output_name[:len(tmp_folder_path)], tmp_folder_path)
if __name__ == '__main__':
unittest.main()
tests/test_main.py
View file @ 5174c719
import unittest
import shutil
import os
# from CpicImgSim import observation_simulation, quick_run
from csst_cpic_sim.main import main, quick_run_v2, observation_simulation_from_config
from csst_cpic_sim.config import config
import os
test_dir = os.path.dirname(__file__)
cases = os.path.join(test_dir, 'testcases')
output = os.path.join(test_dir, 'test_output')
config['output'] = output
from unittest.mock import patch
# from unittest.mock import patch
a = []
import io
import shutil
def clear_output():
if os.path.exists(output):
shutil.rmtree(output)
os.mkdir(output)
class TestMain(unittest.TestCase):
# def test_help(self):
# main(argv = None)
......@@ -36,8 +38,6 @@ class TestMain(unittest.TestCase):
file = os.listdir(output)
self.assertEqual(len(file), 1)
self.assertEqual(file[0][:9], 'demo_0_20')
def test_quick_run_func(self):
clear_output()
......@@ -47,8 +47,6 @@ class TestMain(unittest.TestCase):
file = os.listdir(output)
self.assertEqual(len(file), 1)
self.assertEqual(file[0][:5], 'blank')
def test_full_run_func(self):
clear_output()
......@@ -66,7 +64,7 @@ class TestMain(unittest.TestCase):
file = os.listdir(output)
self.assertEqual(len(file), 4)
clear_output()
observation_simulation_from_config(
os.path.join(cases, '05_sci.yaml'),
......@@ -75,7 +73,6 @@ class TestMain(unittest.TestCase):
file = os.listdir(output)
self.assertEqual(len(file), 1)
self.assertEqual(file[0][:5], 'SCI')
def test_full_run_cmdline(self):
clear_output()
......@@ -83,7 +80,6 @@ class TestMain(unittest.TestCase):
file = os.listdir(output)
self.assertEqual(len(file), 1)
self.assertEqual(file[0][:5], 'SCI')
# def test_main(self):
# target_example = {
......
tests/test_optics.py
View file @ 5174c719
import unittest
import time
from csst_cpic_sim.target import star_photlam
from csst_cpic_sim.optics import ideal_focus_image, focal_convolve, focal_mask, filter_throughput, ideal_focus_image
from csst_cpic_sim.config import which_focalplane, S
from csst_cpic_sim.config import S
import numpy as np
from astropy.io import fits
from csst_cpic_sim.optics import FILTERS
def make_test_sub_image(size, shape):
shape = np.array([shape, shape])
sub_image = np.zeros(shape)
......@@ -28,21 +25,21 @@ def make_test_sub_image(size, shape):
sub_image2 = (np.sqrt(xx**2*2 + yy**2) < size).astype(int)
sub_image = sub_image2 * (1 - sub_image)
return sub_image
def gaussian_psf(band, spectrum, shape, error=0.1):
psf_shape = [shape, shape]
xx, yy = np.mgrid[0:psf_shape[0], 0:psf_shape[1]]
center = np.array([(psf_shape[0]-1)/2, (psf_shape[1]-1)/2])
sigma = 10
psf = np.exp(-((xx-center[0])**2 +
(yy-center[1])**2) / (2*sigma**2))
psf = np.exp(-((xx-center[0])**2 + (yy-center[1])**2) / (2*sigma**2))
psf = psf / psf.sum()
filter = filter_throughput(band)
return psf * (spectrum * filter).integrate()
class TestOptics(unittest.TestCase):
def test_filter_throughtput(self):
......@@ -53,10 +50,6 @@ class TestOptics(unittest.TestCase):
bandpass = filter_throughput('deFault')
self.assertIsInstance(bandpass, S.spectrum.SpectralElement)
# def test_which_focalpalne(self):
# self.assertEqual(which_focalplane('f565'), 'vis')
def test_ideal_focus_image(self):
targets = [
[-20, 0, star_photlam(2, 'G2'), None],
......@@ -64,8 +57,8 @@ class TestOptics(unittest.TestCase):
[8, 0, star_photlam(-5, 'G2'), make_test_sub_image(10, 100)],
]
bandpass = S.Box(6000, 500)
foc = ideal_focus_image(
ideal_focus_image(
bandpass,
targets,
0.0165,
......@@ -73,7 +66,7 @@ class TestOptics(unittest.TestCase):
)
# fits.writeto('foc.fits', foc, overwrite=True)
foc = ideal_focus_image(
ideal_focus_image(
bandpass,
targets,
0.0165,
......@@ -81,7 +74,7 @@ class TestOptics(unittest.TestCase):
rotation=30,
)
foc = ideal_focus_image(
ideal_focus_image(
bandpass,
{},
0.0165,
......@@ -89,7 +82,6 @@ class TestOptics(unittest.TestCase):
rotation=30,
)
# fits.writeto('foc_rot30.fits', foc, overwrite=True)
def test_focal_mask(self):
image = np.zeros((100, 100)) + 1
......@@ -103,28 +95,17 @@ class TestOptics(unittest.TestCase):
[8, 0, star_photlam(-5, 'G2'), make_test_sub_image(10, 100)],
]
img_final = focal_convolve('f661', {})
focal_convolve('f661', {})
img_final = focal_convolve('f661', targets)
focal_convolve('f661', targets)
img_final = focal_convolve('f661', targets, init_shifts=[10, 10])
# fits.writeto('cov.fits', img_final, overwrite=True)
focal_convolve('f661', targets, init_shifts=[10, 10])
# if __name__ == '__main__':
# # # unittest.main()
# # import time
# # from CpicImgSim.target import star_photlam
# test_convolve_psf()
# # import matplotlib.pyplot as plt
......
tests/test_target.py
View file @ 5174c719
......@@ -9,9 +9,10 @@ from csst_cpic_sim.config import S
tests_folder = os.path.dirname(os.path.abspath(__file__))
class TestTarget(unittest.TestCase):
def test_target_object(self):
d_cstar= {
d_cstar = {
'magnitude': 5,
'ra': '120d',
'dec': '40d',
......@@ -34,20 +35,19 @@ class TestTarget(unittest.TestCase):
'phase_angle': 90,
'sp_model': 'hybrid_planet'
}
old_planet = TargetOjbect(d_planet, cstar = cstar)
old_planet = TargetOjbect(d_planet, cstar=cstar)
self.assertEqual(old_planet.sp_model, 'hybrid_planet')
d_planet['sp_model'] = 'bcc_planet'
d_planet['coe_cloud'] = 1
d_planet['coe_metal'] = 0
old_planet = TargetOjbect(d_planet, cstar = cstar)
old_planet = TargetOjbect(d_planet, cstar=cstar)
self.assertEqual(old_planet.sp_model, 'bcc_planet')
def test_bcc_albedo_spectrum(self):
spectrum = AlbedoCat(90, 1, 0)
self.assertIsInstance(spectrum, S.spectrum.SpectralElement)
spectrum = bcc_spectrum(0.5, 0.5)
self.assertIsInstance(spectrum, S.spectrum.SpectralElement)
self.assertEqual(spectrum.waveunits.name, 'angstrom')
......@@ -56,7 +56,6 @@ class TestTarget(unittest.TestCase):
# plt.plot(spectrum.wave, spectrum.throughput)
# plt.show()
def test_hybrid_albedo_spectrum(self):
planet = hybrid_albedo_spectrum(0.5, 1)
self.assertIsInstance(planet, S.spectrum.SpectralElement)
......@@ -124,7 +123,7 @@ class TestTarget(unittest.TestCase):
self.assertRaises(ValueError, extract_target_x_y, dict(ra='120d'))
self.assertRaises(ValueError, extract_target_x_y,
dict(ra='120d', dec='40d'))
def test_detect_template_path(self):
for f in os.listdir():
if os.path.isfile(f):
......@@ -138,19 +137,16 @@ class TestTarget(unittest.TestCase):
self.assertRaises(FileExistsError, detect_template_path, 'demo_5_35.yaml')
def test_target_file_load(self):
t0 = target_file_load({0:0})
t0 = target_file_load({0: 0})
self.assertEqual(t0[0], 0)
t_error = target_file_load(['1'])
self.assertEqual(t_error, {})
t1 = target_file_load('')
self.assertEqual(t1, {})
t1 = target_file_load(' ')
self.assertEqual(t1, {})
......@@ -172,7 +168,6 @@ class TestTarget(unittest.TestCase):
self.assertEqual(t3['cstar']['sp_model'], 'reference')
self.assertEqual(len(t3['objects']), 0)
t4 = target_file_load('demo_0_20')
self.assertEqual(t4['cstar']['magnitude'], 0)
self.assertEqual(t4['cstar']['sp_model'], 'reference')
......@@ -222,4 +217,3 @@ class TestTarget(unittest.TestCase):
template_star = target_file_load('demo_std_star')
spectrums = spectrum_generator(template_star)
self.assertEqual(len(spectrums), 1)
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