Commit 3351be22 authored by Wei Chengliang's avatar Wei Chengliang
Browse files

update codestyle-PEP8

parent 8fd9723d
......@@ -98,7 +98,7 @@ class Quasar(MockObject):
del self.sed
def getGSObj(self, psf, g1=0, g2=0, flux=None, filt=None, tel=None, exptime=150.):
if flux == None:
if flux is None:
flux = self.getElectronFluxFilt(filt, tel, exptime)
qso = galsim.Gaussian(sigma=1.e-8, flux=1.)
qso = qso.withFlux(flux)
......
......@@ -21,7 +21,7 @@ class Stamp(MockObject):
del self.sed
def drawObj_multiband(self, tel, pos_img, psf_model, bandpass_list, filt, chip, nphotons_tot=None, g1=0, g2=0, exptime=150., fd_shear=None):
if nphotons_tot == None:
if nphotons_tot is None:
nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime)
try:
......
......@@ -22,7 +22,7 @@ class Star(MockObject):
del self.sed
def getGSObj(self, psf, g1=0, g2=0, flux=None, filt=None, tel=None, exptime=150.):
if flux == None:
if flux is None:
flux = self.getElectronFluxFilt(filt, tel, exptime)
# star = galsim.Gaussian(sigma=1.e-8, flux=1.)
star = galsim.DeltaFunction()
......@@ -35,7 +35,7 @@ class Star(MockObject):
raise ValueError(
"!!!The number of PSF profiles and the number of bandpasses must be equal.")
objs = []
if nphotons_tot == None:
if nphotons_tot is None:
nphotons_tot = self.getElectronFluxFilt(filt, tel, exptime)
try:
......
......@@ -428,8 +428,8 @@ class TestSpecDisperse(unittest.TestCase):
plt.xlim(6200, 10000)
plt.ylim(1, 4)
plt.yscale('log')
plt.xlabel('$\lambda$')
plt.ylabel('$F\lambda$')
plt.xlabel(r'$\lambda$')
plt.ylabel(r'$F\lambda$')
plt.legend(['one spec', 'split in 8000 A'])
plt.show()
......
......@@ -192,7 +192,7 @@ class TestStraylight(unittest.TestCase):
plt.figure()
plt.plot(spec['WAVELENGTH'], spec['FLUX'], 'r')
plt.xlabel('WAVELENGTH')
plt.ylabel('F$\lambda$(erg/s/cm2/A/arcsec2)')
plt.ylabel(r'F$\lambda$(erg/s/cm2/A/arcsec2)')
plt.xlim(2000, 10000)
plt.show()
median = np.median(sl_e_pix[0:tnum])
......
......@@ -10,11 +10,11 @@ import yaml
from observation_sim.instruments import Chip, Filter, FilterParam, FocalPlane
### test FUNCTION --- START ###
# test FUNCTION --- START ###
def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150., InputDark=None):
if InputDark == None:
if InputDark is 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)
......@@ -31,7 +31,7 @@ def get_base_img(img, chip, read_noise, readout_time, dark_noise, exptime=150.,
base_img2[ny:, :] = arr[::-1, :]
base_img2[:, :] = base_img2[:, :]*(readout_time/ny)*dark_noise
return base_img1+base_img2
### test FUNCTION --- END ###
# test FUNCTION --- END ###
def defineCCD(iccd, config_file):
......
......@@ -221,8 +221,8 @@ def ccdLayout(xpixTar, ypixTar, figout="ccdLayout.pdf"):
ax.text(ix0+500, iy0+1500, "%s#%s" %
(fType, ischip), fontsize=12, color="grey")
ax.plot(xpixTar, ypixTar, "r*", ms=12)
ax.set_xlabel("$X\,[\mathrm{pixels}]$", fontsize=20)
ax.set_ylabel("$Y\,[\mathrm{pixels}]$", fontsize=20)
ax.set_xlabel(r"$X\,[\mathrm{pixels}]$", fontsize=20)
ax.set_ylabel(r"$Y\,[\mathrm{pixels}]$", fontsize=20)
ax.invert_yaxis()
ax.axis('off')
plt.savefig(figout)
......
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