From 2497d9d13fdcdf0169da68c7be8ce71b2752a440 Mon Sep 17 00:00:00 2001 From: Shuai Feng <451424498@qq.com> Date: Sat, 26 Oct 2024 22:54:19 +0800 Subject: [PATCH] add PEP8 bugs --- tests/test_cube3d.py | 53 +++++----- tests/test_map2d.py | 118 ++++++++++++----------- tests/test_spec1d.py | 223 ++++++++++++++++++++++--------------------- 3 files changed, 200 insertions(+), 194 deletions(-) diff --git a/tests/test_cube3d.py b/tests/test_cube3d.py index cdd1587..a807641 100644 --- a/tests/test_cube3d.py +++ b/tests/test_cube3d.py @@ -1,14 +1,16 @@ import sys -sys.path.insert(0, '..') import unittest from csst_ifs_gehong import spec1d as s from csst_ifs_gehong import map2d as m from csst_ifs_gehong import cube3d as b from csst_ifs_gehong import config as c import numpy as np -import matplotlib.pyplot as plt +# import matplotlib.pyplot as plt + +# import warnings + +sys.path.insert(0, '..') -import warnings class test_cube3d(unittest.TestCase): @@ -25,7 +27,7 @@ class test_cube3d(unittest.TestCase): print(" ") inst = c.config() - gas_tem = s.EmissionLineTemplate(inst, model = 'hii') + gas_tem = s.EmissionLineTemplate(inst, model='hii') stellar_tem = s.StellarContinuumTemplate(inst) sbmap = m.Map2d(inst) @@ -35,14 +37,14 @@ class test_cube3d(unittest.TestCase): vdispmap = m.Map2d(inst) vdispmap.gred_map() agemap = m.Map2d(inst) - agemap.gred_map(a0 = 9.5, gred = -1.2) + agemap.gred_map(a0=9.5, gred=-1.2) fehmap = m.Map2d(inst) - fehmap.gred_map(a0 = -0.2, gred = -0.1) + fehmap.gred_map(a0=-0.2, gred=-0.1) ebvmap = m.Map2d(inst) - ebvmap.gred_map(a0 = 0.2, gred = -0.1) - stellarcontinuum = m.StellarPopulationMap(inst, sbright = sbmap, logage = agemap, - feh = fehmap, vel = velmap, - vdisp = vdispmap, ebv = ebvmap) + ebvmap.gred_map(a0=0.2, gred=-0.1) + stellarcontinuum = m.StellarPopulationMap(inst, sbright=sbmap, logage=agemap, + feh=fehmap, vel=velmap, + vdisp=vdispmap, ebv=ebvmap) sbmap = m.Map2d(inst) sbmap.sersic_map() @@ -51,32 +53,33 @@ class test_cube3d(unittest.TestCase): vdispmap = m.Map2d(inst) vdispmap.gred_map() agemap = m.Map2d(inst) - agemap.gred_map(a0 = 9.5, gred = -1.2) + agemap.gred_map(a0=9.5, gred=-1.2) fehmap = m.Map2d(inst) - fehmap.gred_map(a0 = -0.2, gred = -0.1) + fehmap.gred_map(a0=-0.2, gred=-0.1) ebvmap = m.Map2d(inst) - ebvmap.gred_map(a0 = 0.2, gred = -0.1) - ionizedgas = m.IonizedGasMap(inst, halpha = agemap, zh = fehmap, - vel = velmap, vdisp = vdispmap, ebv = ebvmap) + ebvmap.gred_map(a0=0.2, gred=-0.1) + ionizedgas = m.IonizedGasMap(inst, halpha=agemap, zh=fehmap, + vel=velmap, vdisp=vdispmap, ebv=ebvmap) - u = b.Cube3D(inst, stellar_map = stellarcontinuum, gas_map = ionizedgas) + u = b.Cube3D(inst, stellar_map=stellarcontinuum, gas_map=ionizedgas) - u.make_cube(stellar_tem = stellar_tem, hii_tem = gas_tem) + u.make_cube(stellar_tem=stellar_tem, hii_tem=gas_tem) u.savefits('result.fits') - if np.sum(u.flux[15,15,:]) != np.sum(u.flux[5,25,:]): + if np.sum(u.flux[15, 15, :]) != np.sum(u.flux[5, 25, :]): print("No Problem!") - #plt.figure(figsize=(16,4)) - #plt.plot(u.wave,np.log10(u.flux[50,50,:]),color='blue',label=r'$v=%4.2f$'%(velmap.map[50,50])) - #plt.plot(u.wave,np.log10(u.flux[5,95,:]),color='red',label=r'$v=%4.2f$'%(velmap.map[5,95])) - #plt.xlim(3500,9500) - #plt.legend(frameon=False) - #plt.savefig('./image/test_cube3d_spec.png') + # plt.figure(figsize=(16,4)) + # plt.plot(u.wave,np.log10(u.flux[50,50,:]),color='blue',label=r'$v=%4.2f$'%(velmap.map[50,50])) + # plt.plot(u.wave,np.log10(u.flux[5,95,:]),color='red',label=r'$v=%4.2f$'%(velmap.map[5,95])) + # plt.xlim(3500,9500) + # plt.legend(frameon=False) + # plt.savefig('./image/test_cube3d_spec.png') print("-----------------------------------------------------------------") print(">>> Unit test for **Spec Cube** Modelling has been passed!") -if __name__=='__main__': + +if __name__ == '__main__': unittest.main() diff --git a/tests/test_map2d.py b/tests/test_map2d.py index 6e19be3..a24d1b6 100644 --- a/tests/test_map2d.py +++ b/tests/test_map2d.py @@ -1,12 +1,13 @@ import sys -sys.path.insert(0, '..') import unittest from csst_ifs_gehong import map2d as m from csst_ifs_gehong import config as c import numpy as np -import matplotlib.pyplot as plt +# import matplotlib.pyplot as plt +# import warnings + +sys.path.insert(0, '..') -import warnings class test_map2d(unittest.TestCase): @@ -22,44 +23,44 @@ class test_map2d(unittest.TestCase): print(" ") print("Pars: Total Magnitude") sbmap0 = m.Map2d(inst) - sbmap0.sersic_map(mag = 12, r_eff = 2, n = 2.5, ellip = 0.5, theta = -50) + sbmap0.sersic_map(mag=12, r_eff=2, n=2.5, ellip=0.5, theta=-50) sbmap1 = m.Map2d(inst) - sbmap1.sersic_map(mag = 14, r_eff = 2, n = 2.5, ellip = 0.5, theta = -50) + sbmap1.sersic_map(mag=14, r_eff=2, n=2.5, ellip=0.5, theta=-50) if np.sum(sbmap1.map) != np.sum(sbmap0.map): print("No Problem!") print("Pars: Effective Radius") sbmap1 = m.Map2d(inst) - sbmap1.sersic_map(mag = 12, r_eff = 4, n = 2.5, ellip = 0.5, theta = -50) + sbmap1.sersic_map(mag=12, r_eff=4, n=2.5, ellip=0.5, theta=-50) if np.sum(sbmap1.map) != np.sum(sbmap0.map): print("No Problem!") print("Pars: Sersic Index") sbmap1 = m.Map2d(inst) - sbmap1.sersic_map(mag = 12, r_eff = 2, n = 5, ellip = 0.5, theta = -50) + sbmap1.sersic_map(mag=12, r_eff=2, n=5, ellip=0.5, theta=-50) if np.sum(sbmap1.map) != np.sum(sbmap0.map): print("No Problem!") print("Pars: Ellipse") sbmap1 = m.Map2d(inst) - sbmap1.sersic_map(mag = 12, r_eff = 2, n = 2.5, ellip = 0.2, theta = -50) + sbmap1.sersic_map(mag=12, r_eff=2, n=2.5, ellip=0.2, theta=-50) if np.sum(sbmap1.map) != np.sum(sbmap0.map): print("No Problem!") print("Pars: Position Angle") sbmap1 = m.Map2d(inst) - sbmap1.sersic_map(mag = 12, r_eff = 2, n = 2.5, ellip = 0.5, theta = 0) + sbmap1.sersic_map(mag=12, r_eff=2, n=2.5, ellip=0.5, theta=0) if np.sum(sbmap1.map) != np.sum(sbmap0.map): print("No Problem!") - #plt.figure(figsize=(6,5)) - #plt.imshow(sbmap0.map, origin='lower', cmap='gray_r') - #plt.xticks([]) - #plt.yticks([]) - #plt.title('Surface Brightness Modelling') - #plt.colorbar(label='mag') - #plt.tight_layout() - #plt.savefig('./image/test_map2d_SBmap.png') + # plt.figure(figsize=(6,5)) + # plt.imshow(sbmap0.map, origin='lower', cmap='gray_r') + # plt.xticks([]) + # plt.yticks([]) + # plt.title('Surface Brightness Modelling') + # plt.colorbar(label='mag') + # plt.tight_layout() + # plt.savefig('./image/test_map2d_SBmap.png') def test_velmap(self): @@ -69,38 +70,38 @@ class test_map2d(unittest.TestCase): print(" ") print("Pars: Rotation Velocity") velmap0 = m.Map2d(inst) - velmap0.tanh_map(vmax = 200, rt = 2, ellip = 0.5, theta = -50) + velmap0.tanh_map(vmax=200, rt=2, ellip=0.5, theta=-50) velmap1 = m.Map2d(inst) - velmap1.tanh_map(vmax = 100, rt = 2, ellip = 0.5, theta = -50) + velmap1.tanh_map(vmax=100, rt=2, ellip=0.5, theta=-50) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") print("Pars: Turnover Radius") velmap1 = m.Map2d(inst) - velmap1.tanh_map(vmax = 200, rt = 3, ellip = 0.5, theta = -50) + velmap1.tanh_map(vmax=200, rt=3, ellip=0.5, theta=-50) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") print("Pars: Inclination Angle") velmap1 = m.Map2d(inst) - velmap1.tanh_map(vmax = 200, rt = 2, ellip = 0.3, theta = -50) + velmap1.tanh_map(vmax=200, rt=2, ellip=0.3, theta=-50) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") print("Pars: Position Angle") velmap1 = m.Map2d(inst) - velmap1.tanh_map(vmax = 200, rt = 2, ellip = 0.5, theta = 100) + velmap1.tanh_map(vmax=200, rt=2, ellip=0.5, theta=100) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") - #plt.figure(figsize=(6,5)) - #plt.imshow(velmap0.map, origin='lower', cmap='RdBu_r') - #plt.xticks([]) - #plt.yticks([]) - #plt.title('Velocity Map Modelling') - #plt.colorbar(label='km/s') - #plt.tight_layout() - #plt.savefig('./image/test_map2d_velmap.png') + # plt.figure(figsize=(6,5)) + # plt.imshow(velmap0.map, origin='lower', cmap='RdBu_r') + # plt.xticks([]) + # plt.yticks([]) + # plt.title('Velocity Map Modelling') + # plt.colorbar(label='km/s') + # plt.tight_layout() + # plt.savefig('./image/test_map2d_velmap.png') def test_gremap(self): @@ -110,44 +111,44 @@ class test_map2d(unittest.TestCase): print(" ") print("Pars: Central Intensity") velmap0 = m.Map2d(inst) - velmap0.gred_map(a0 = 10, r_eff = 1, gred = -1, ellip = 0.5, theta = 0) + velmap0.gred_map(a0=10, r_eff=1, gred=-1, ellip=0.5, theta=0) velmap1 = m.Map2d(inst) - velmap1.gred_map(a0 = 20, r_eff = 1, gred = -1, ellip = 0.5, theta = 0) + velmap1.gred_map(a0=20, r_eff=1, gred=-1, ellip=0.5, theta=0) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") print("Pars: Effective Radius") velmap1 = m.Map2d(inst) - velmap1.gred_map(a0 = 10, r_eff = 2, gred = -1, ellip = 0.5, theta = 0) + velmap1.gred_map(a0=10, r_eff=2, gred=-1, ellip=0.5, theta=0) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") - + print("Pars: Grediant") velmap1 = m.Map2d(inst) - velmap1.gred_map(a0 = 10, r_eff = 1, gred = -2, ellip = 0.5, theta = 0) + velmap1.gred_map(a0=10, r_eff=1, gred=-2, ellip=0.5, theta=0) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") print("Pars: Inclination Angle") velmap1 = m.Map2d(inst) - velmap1.gred_map(a0 = 10, r_eff = 1, gred = -1, ellip = 0.3, theta = 0) + velmap1.gred_map(a0=10, r_eff=1, gred=-1, ellip=0.3, theta=0) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") print("Pars: Position Angle") velmap1 = m.Map2d(inst) - velmap1.gred_map(a0 = 10, r_eff = 1, gred = -1, ellip = 0.5, theta = 100) + velmap1.gred_map(a0=10, r_eff=1, gred=-1, ellip=0.5, theta=100) if np.sum(velmap1.map) != np.sum(velmap0.map): print("No Problem!") - #plt.figure(figsize=(6,5)) - #plt.imshow(velmap0.map, origin='lower', cmap='RdBu_r') - #plt.xticks([]) - #plt.yticks([]) - #plt.title('Gredient Map Modelling') - #plt.colorbar(label='') - #plt.tight_layout() - #plt.savefig('./image/test_map2d_gredmap.png') + # plt.figure(figsize=(6,5)) + # plt.imshow(velmap0.map, origin='lower', cmap='RdBu_r') + # plt.xticks([]) + # plt.yticks([]) + # plt.title('Gredient Map Modelling') + # plt.colorbar(label='') + # plt.tight_layout() + # plt.savefig('./image/test_map2d_gredmap.png') def test_input_image(self): @@ -156,7 +157,7 @@ class test_map2d(unittest.TestCase): print("------------------------ Feed Image------------------------") print(" ") velmap0 = m.Map2d(inst) - velmap0.gred_map(a0 = 10, r_eff = 1, gred = -1, ellip = 0.5, theta = 0) + velmap0.gred_map(a0=10, r_eff=1, gred=-1, ellip=0.5, theta=0) velmap1 = m.Map2d(inst) velmap1.load_map(velmap0.map) if np.sum(velmap1.map) == np.sum(velmap0.map): @@ -176,17 +177,17 @@ class test_map2d(unittest.TestCase): vdispmap.gred_map() agemap = m.Map2d(inst) - agemap.gred_map(a0 = 9.5, gred = -1.2) + agemap.gred_map(a0=9.5, gred=-1.2) fehmap = m.Map2d(inst) - fehmap.gred_map(a0 = -0.2, gred = -0.1) + fehmap.gred_map(a0=-0.2, gred=-0.1) ebvmap = m.Map2d(inst) - ebvmap.gred_map(a0 = 0.2, gred = -0.1) + ebvmap.gred_map(a0=0.2, gred=-0.1) + + stellarcontinuum = m.StellarPopulationMap(inst, sbright=sbmap, logage=agemap, feh=fehmap, + vel=velmap, vdisp=vdispmap, ebv=ebvmap) - stellarcontinuum = m.StellarPopulationMap(inst, sbright = sbmap, logage = agemap, feh = fehmap, - vel = velmap, vdisp = vdispmap, ebv = ebvmap) - if np.sum(stellarcontinuum.sbright) == np.sum(sbmap.map): print("SBmap No Problem!") if np.sum(stellarcontinuum.age) == np.sum(agemap.map): @@ -214,14 +215,14 @@ class test_map2d(unittest.TestCase): vdispmap.gred_map() logohmap = m.Map2d(inst) - logohmap.gred_map(a0 = 9.5, gred = -1.2) + logohmap.gred_map(a0=9.5, gred=-1.2) ebvmap = m.Map2d(inst) - ebvmap.gred_map(a0 = 0.2, gred = -0.1) + ebvmap.gred_map(a0=0.2, gred=-0.1) + + gas = m.IonizedGasMap(inst, halpha=halphamap, zh=logohmap, + vel=velmap, vdisp=vdispmap, ebv=ebvmap) - gas = m.IonizedGasMap(inst, halpha = halphamap, zh = logohmap, - vel = velmap, vdisp = vdispmap, ebv = ebvmap) - if np.sum(gas.halpha) == np.sum(halphamap.map): print("Halphamap No Problem!") if np.sum(gas.zh) == np.sum(logohmap.map): @@ -233,5 +234,6 @@ class test_map2d(unittest.TestCase): if np.sum(gas.ebv) == np.sum(ebvmap.map): print("SBmap No Problem!") -if __name__=='__main__': + +if __name__ == '__main__': unittest.main() diff --git a/tests/test_spec1d.py b/tests/test_spec1d.py index 94d97ba..501a601 100644 --- a/tests/test_spec1d.py +++ b/tests/test_spec1d.py @@ -1,13 +1,13 @@ import sys -sys.path.insert(0, '..') import unittest from csst_ifs_gehong import spec1d as s from csst_ifs_gehong import config as c import numpy as np import matplotlib.pyplot as plt - import warnings warnings.simplefilter("ignore", ResourceWarning) +sys.path.insert(0, '..') + class test_spec1d(unittest.TestCase): @@ -27,89 +27,89 @@ class test_spec1d(unittest.TestCase): temp = s.StellarContinuumTemplate(inst) - plt.figure(figsize=(15,12)) + plt.figure(figsize=(15, 12)) print("------------------------Test Magnitude---------------------------") print("1. mr=15") - sed1 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0, vel = 100, vdisp = 100, ebv = 0) + sed1 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0, vel=100, vdisp=100, ebv=0) print("2. mr=13") - sed2 = s.StellarContinuum(inst, temp, mag = 13, age = 1, feh = 0, vel = 100, vdisp = 100, ebv = 0) + sed2 = s.StellarContinuum(inst, temp, mag=13, age=1, feh=0, vel=100, vdisp=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(231) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'mr = 15') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'mr = 13') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='mr = 15') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='mr = 13') plt.legend(frameon=False) plt.xlim(3900, 6200) - + print("---------------------------Test Age-----------------------------") print("1. Age=0.1Gyr") - sed1 = s.StellarContinuum(inst, temp, mag = 15, age = 0.1, feh = 0, vel = 100, vdisp = 100, ebv = 0) + sed1 = s.StellarContinuum(inst, temp, mag=15, age=0.1, feh=0, vel=100, vdisp=100, ebv=0) print("2. Age=1Gyr") - sed2 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0, vel = 100, vdisp = 100, ebv = 0) + sed2 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0, vel=100, vdisp=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(232) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'Age = 0.1Gyr') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'Age = 1Gyr') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='Age = 0.1Gyr') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='Age = 1Gyr') plt.legend(frameon=False) plt.xlim(3900, 6200) print("-----------------------Test Metallicity-------------------------") print("1. [Fe/H]=0.5") - sed1 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0.5, vel = 100, vdisp = 100, ebv = 0) + sed1 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0.5, vel=100, vdisp=100, ebv=0) print("2. [Fe/H]=-0.5") - sed2 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = -0.5, vel = 100, vdisp = 100, ebv = 0) + sed2 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=-0.5, vel=100, vdisp=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(233) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = '[Fe/H] = 0.5') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = '[Fe/H] = -0.5') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='[Fe/H] = 0.5') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='[Fe/H] = -0.5') plt.legend(frameon=False) plt.xlim(3900, 6200) - + print("------------------------Test Velocity---------------------------") print("1. v=3000km/s") - sed1 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0, vel = 3000, vdisp = 100, ebv = 0) + sed1 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0, vel=3000, vdisp=100, ebv=0) print("2. v=6000km/s") - sed2 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0, vel = 6000, vdisp = 100, ebv = 0) + sed2 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0, vel=6000, vdisp=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(234) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'v = 3000km/s') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'v = 6000km/s') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='v = 3000km/s') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='v = 6000km/s') plt.legend(frameon=False) plt.xlim(3900, 6200) - + print("-------------------Test Velocity Dispersion----------------------") print("1. sigma=180km/s") - sed1 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0.5, vel = 100, vdisp = 180, ebv = 0) + sed1 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0.5, vel=100, vdisp=180, ebv=0) print("2. sigma=350km/s") - sed2 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0.5, vel = 100, vdisp = 350, ebv = 0) + sed2 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0.5, vel=100, vdisp=350, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(235) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'sigma = 180km/s') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'sigma = 350km/s') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='sigma = 180km/s') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='sigma = 350km/s') plt.legend(frameon=False) plt.xlim(3900, 6200) print("----------------------Test Dust Extinction-----------------------") print("1. E(B-V)=0.1") - sed1 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0.5, vel = 100, vdisp = 100, ebv = 0.1) + sed1 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0.5, vel=100, vdisp=100, ebv=0.1) print("2. E(B-V)=0.3") - sed2 = s.StellarContinuum(inst, temp, mag = 15, age = 1, feh = 0.5, vel = 100, vdisp = 100, ebv = 0.3) + sed2 = s.StellarContinuum(inst, temp, mag=15, age=1, feh=0.5, vel=100, vdisp=100, ebv=0.3) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(236) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'E(B-V) = 0.1') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'E(B-V) = 0.3') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='E(B-V) = 0.1') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='E(B-V) = 0.3') plt.legend(frameon=False) plt.xlim(3900, 6200) plt.tight_layout() - #plt.savefig('./image/test_spec1d_stellarpopulation.png') - + # plt.savefig('./image/test_spec1d_stellarpopulation.png') + def test_Star(self): inst = c.config() @@ -122,62 +122,62 @@ class test_spec1d(unittest.TestCase): temp = s.SingleStarTemplate(inst) - plt.figure(figsize=(15,12)) + plt.figure(figsize=(15, 12)) print("------------------------Test Magnitude---------------------------") print("1. mr=15") - sed1 = s.SingleStar(inst, temp, mag = 15, teff = 10000, feh = 0, vel = 100, ebv = 0) + sed1 = s.SingleStar(inst, temp, mag=15, teff=10000, feh=0, vel=100, ebv=0) print("2. mr=13") - sed2 = s.SingleStar(inst, temp, mag = 13, teff = 10000, feh = 0, vel = 100, ebv = 0) + sed2 = s.SingleStar(inst, temp, mag=13, teff=10000, feh=0, vel=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(221) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'mr = 15') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'mr = 13') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='mr = 15') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='mr = 13') plt.legend(frameon=False) plt.xlim(3900, 6200) print("------------------------Test Tempreture--------------------------") print("1. Teff=6000K") - sed1 = s.SingleStar(inst, temp, mag = 15, teff = 6000, feh = 0, vel = 100, ebv = 0) + sed1 = s.SingleStar(inst, temp, mag=15, teff=6000, feh=0, vel=100, ebv=0) print("1. Teff=10000K") - sed2 = s.SingleStar(inst, temp, mag = 15, teff = 10000, feh = 0, vel = 100, ebv = 0) + sed2 = s.SingleStar(inst, temp, mag=15, teff=10000, feh=0, vel=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(222) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'Teff = 6000K') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'Teff = 10000K') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='Teff = 6000K') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='Teff = 10000K') plt.legend(frameon=False) plt.xlim(3900, 6200) print("-----------------------Test Metallicity--------------------------") print("1. [Fe/H]=-0.5") - sed1 = s.SingleStar(inst, temp, mag = 15, teff = 10000, feh = -0.5, vel = 100, ebv = 0) + sed1 = s.SingleStar(inst, temp, mag=15, teff=10000, feh=-0.5, vel=100, ebv=0) print("2. [Fe/H]=0.5") - sed2 = s.SingleStar(inst, temp, mag = 15, teff = 10000, feh = 0.5, vel = 100, ebv = 0) + sed2 = s.SingleStar(inst, temp, mag=15, teff=10000, feh=0.5, vel=100, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(223) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = '[Fe/H] = -0.5') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = '[Fe/H] = 0.5') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='[Fe/H] = -0.5') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='[Fe/H] = 0.5') plt.legend(frameon=False) plt.xlim(3900, 6200) print("------------------------Test Velocity---------------------------") print("1. v=-200km/s") - sed1 = s.SingleStar(inst, temp, mag = 15, teff = 10000, feh = 0, vel = -200, ebv = 0) + sed1 = s.SingleStar(inst, temp, mag=15, teff=10000, feh=0, vel=-200, ebv=0) print("2. v=200km/s") - sed2 = s.SingleStar(inst, temp, mag = 15, teff = 10000, feh = 0, vel = 200, ebv = 0) + sed2 = s.SingleStar(inst, temp, mag=15, teff=10000, feh=0, vel=200, ebv=0) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(224) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'v = -200km/s') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'v = 200km/s') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='v = -200km/s') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='v = 200km/s') plt.legend(frameon=False) plt.xlim(3900, 6200) - + plt.tight_layout() - #plt.savefig('./image/test_spec1d_singlestar.png') + # plt.savefig('./image/test_spec1d_singlestar.png') print("-----------------------------------------------------------------") print(">>> Unit test for **Single Stellar** Modelling has been passed!") @@ -192,82 +192,82 @@ class test_spec1d(unittest.TestCase): print(">>> Start Test") print(" ") - temp = s.EmissionLineTemplate(inst, model = 'hii') + temp = s.EmissionLineTemplate(inst, model='hii') - plt.figure(figsize=(15,12)) + plt.figure(figsize=(15, 12)) print("---------------------Test Total Halpha Flux----------------------") print("1. F_Halpha = 10 * 1e-17 erg/s/cm^2") - sed1 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 100, vdisp = 120, ebv = 0.1) + sed1 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=100, vdisp=120, ebv=0.1) print("2. F_Halpha = 50 * 1e-17 erg/s/cm^2") - sed2 = s.HII_Region(inst, temp, halpha = 50, logz = -1, vel = 100, vdisp = 120, ebv = 0.1) + sed2 = s.HII_Region(inst, temp, halpha=50, logz=-1, vel=100, vdisp=120, ebv=0.1) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(231) - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'Flux = 50') - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'Flux = 10') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='Flux = 50') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='Flux = 10') plt.legend(frameon=False) plt.xlim(3900, 6200) print("-------------------Test Gas-Phase Metallicity--------------------") print("1. [Z/H] = 0") - sed1 = s.HII_Region(inst, temp, halpha = 10, logz = 0, vel = 100, vdisp = 120, ebv = 0.1) + sed1 = s.HII_Region(inst, temp, halpha=10, logz=0, vel=100, vdisp=120, ebv=0.1) print("2. [Z/H] = -0.5") - sed2 = s.HII_Region(inst, temp, halpha = 10, logz = -0.5, vel = 100, vdisp = 120, ebv = 0.1) + sed2 = s.HII_Region(inst, temp, halpha=10, logz=-0.5, vel=100, vdisp=120, ebv=0.1) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(232) - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = '[Z/H] = -0.5') - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = '[Z/H] = 0') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='[Z/H] = -0.5') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='[Z/H] = 0') plt.legend(frameon=False) plt.xlim(3900, 6200) print("-------------------------Test Velocity--------------------------") print("1. v=3000km/s") - sed1 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 3000, vdisp = 120, ebv = 0.1) + sed1 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=3000, vdisp=120, ebv=0.1) print("2. v=6000km/s") - sed2 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 6000, vdisp = 120, ebv = 0.1) + sed2 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=6000, vdisp=120, ebv=0.1) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(233) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'v = 3000km/s') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'v = 6000km/s') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='v = 3000km/s') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='v = 6000km/s') plt.legend(frameon=False) plt.xlim(3900, 6200) print("--------------------Test Velocity Dispersion---------------------") print("1. sigma=500km/s") - sed1 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 100, vdisp = 500, ebv = 0.1) + sed1 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=100, vdisp=500, ebv=0.1) print("2. sigma=100km/s") - sed2 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 100, vdisp = 100, ebv = 0.1) + sed2 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=100, vdisp=100, ebv=0.1) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(234) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'sigma = 500km/s') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'sigma = 100km/s') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='sigma = 500km/s') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='sigma = 100km/s') plt.legend(frameon=False) plt.xlim(3900, 6200) print("----------------------Test Dust Extinction-----------------------") print("1. E(B-V)=0.1") - sed1 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 100, vdisp = 120, ebv = 0.1) + sed1 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=100, vdisp=120, ebv=0.1) print("2. E(B-V)=0.3") - sed1 = s.HII_Region(inst, temp, halpha = 10, logz = -1, vel = 100, vdisp = 120, ebv = 0.3) + sed1 = s.HII_Region(inst, temp, halpha=10, logz=-1, vel=100, vdisp=120, ebv=0.3) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(235) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'E(B-V) = 0.1') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'E(B-V) = 0.3') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='E(B-V) = 0.1') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='E(B-V) = 0.3') plt.legend(frameon=False) plt.xlim(3900, 6200) plt.tight_layout() - #plt.savefig('./image/test_spec1d_ionizedgas.png') + # plt.savefig('./image/test_spec1d_ionizedgas.png') print("-----------------------------------------------------------------") print(">>> Unit test for **Ionized Gas** Modelling has been passed!") @@ -282,87 +282,88 @@ class test_spec1d(unittest.TestCase): print(">>> Start Test") print(" ") - temp = s.EmissionLineTemplate(inst, model = 'nlr') + temp = s.EmissionLineTemplate(inst, model='nlr') - plt.figure(figsize=(15,12)) + plt.figure(figsize=(15, 12)) print("-------------------------Test Redshift--------------------------") print("1. z=0.2") - sed1 = s.AGN(inst, temp, bhmass = 1e5, edd_ratio = 0.05, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.1, dist = 20) + sed1 = s.AGN(inst, temp, bhmass=1e5, edd_ratio=0.05, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.1, dist=20) print("2. z=0.4") - sed2 = s.AGN(inst, temp, bhmass = 1e5, edd_ratio = 0.05, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 40000, logz = 0, ebv = 0.1, dist = 20) + sed2 = s.AGN(inst, temp, bhmass=1e5, edd_ratio=0.05, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=40000, logz=0, ebv=0.1, dist=20) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(221) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'z = 0.2') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'z = 0.4') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='z = 0.2') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='z = 0.4') plt.legend(frameon=False) plt.xlim(3900, 6200) - + print("----------------------Test Black Hole Mass-----------------------") print("1. Mbh=1e6") - sed1 = s.AGN(inst, temp, bhmass = 1e6, edd_ratio = 0.05, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.1, dist = 20) + sed1 = s.AGN(inst, temp, bhmass=1e6, edd_ratio=0.05, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.1, dist=20) print("2. Mbh=5e6") - sed2 = s.AGN(inst, temp, bhmass = 5e6, edd_ratio = 0.05, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.1, dist = 20) + sed2 = s.AGN(inst, temp, bhmass=5e6, edd_ratio=0.05, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.1, dist=20) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(222) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'Mbh = 1e6') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'Mbh = 5e6') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='Mbh = 1e6') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='Mbh = 5e6') plt.legend(frameon=False) plt.xlim(3900, 6200) print("------------------------Eddington Ratio-------------------------") print("1. Edd Ratio = 1.0") - sed1 = s.AGN(inst, temp, bhmass = 1e5, edd_ratio = 1.0, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.1, dist = 20) + sed1 = s.AGN(inst, temp, bhmass=1e5, edd_ratio=1.0, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.1, dist=20) print("2. Edd Ratio = 0.1") - sed2 = s.AGN(inst, temp, bhmass = 1e5, edd_ratio = 0.1, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.1, dist = 20) + sed2 = s.AGN(inst, temp, bhmass=1e5, edd_ratio=0.1, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.1, dist=20) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(223) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'EddRatio = 1.0') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'EddRatio = 0.1') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='EddRatio = 1.0') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='EddRatio = 0.1') plt.legend(frameon=False) plt.xlim(3900, 6200) print("----------------------Test Dust Extinction-----------------------") print("1. E(B-V)=0.1") - sed1 = s.AGN(inst, temp, bhmass = 1e5, edd_ratio = 0.05, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.1, dist = 20) + sed1 = s.AGN(inst, temp, bhmass=1e5, edd_ratio=0.05, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.1, dist=20) print("2. E(B-V)=0.3") - sed2 = s.AGN(inst, temp, bhmass = 1e5, edd_ratio = 0.05, - halpha_broad = 100, halpha_narrow = 100, vdisp_broad = 2000, vdisp_narrow = 500, - vel = 20000, logz = 0, ebv = 0.3, dist = 20) + sed2 = s.AGN(inst, temp, bhmass=1e5, edd_ratio=0.05, + halpha_broad=100, halpha_narrow=100, vdisp_broad=2000, vdisp_narrow=500, + vel=20000, logz=0, ebv=0.3, dist=20) if np.sum(sed1.flux) != np.sum(sed2.flux): print("No Problem!") plt.subplot(224) - plt.plot(sed1.wave, sed1.flux, color = 'blue', lw = 2, label = 'E(B-V) = 0.1') - plt.plot(sed2.wave, sed2.flux, color = 'red', lw = 2, label = 'E(B-V) = 0.3') + plt.plot(sed1.wave, sed1.flux, color='blue', lw=2, label='E(B-V) = 0.1') + plt.plot(sed2.wave, sed2.flux, color='red', lw=2, label='E(B-V) = 0.3') plt.legend(frameon=False) plt.xlim(3900, 6200) plt.tight_layout() - #plt.savefig('./image/test_spec1d_agn.png') + # plt.savefig('./image/test_spec1d_agn.png') print("-----------------------------------------------------------------") print(">>> Unit test for **AGN** Modelling has been passed!") -if __name__=='__main__': + +if __name__ == '__main__': unittest.main() -- GitLab