def test_pack():
for channel, nrows, ncolumns in ('red',16,32), ('blue',32,64):
obs = PacsSimulation(Pointing((0., 0., 0.), 0.), channel)
for dt in (np.uint8, np.uint16, np.uint32, np.uint64):
a = np.arange(nrows*ncolumns*3, dtype=dt) \
.reshape((nrows,ncolumns,-1))
p = obs.pack(a)
assert all_eq(a[1,0:16,:], p[16:32,:])
u = obs.unpack(p)
assert all_eq(a, u)
def func2(c, b, m):
pointing.time = time * c
if (c,b,m) in ok:
simul = PacsSimulation(pointing, b, mode=m)
assert simul.slice[0].compression_factor == c
assert simul.slice[0].mode == m
else:
try:
simul = PacsSimulation(pointing, b, mode=m)
except ValueError:
pass
else:
print c,b,m
assert False
def test1():
# creation of the sky map
msize = 50
mymap = gaussian(2*(msize*2+1,), 10, unit='Jy/pixel')
cd = np.array([[-1., 0.],[0., 1.]]) / 3600.
header = create_fitsheader(fromdata=mymap, crval=[53.,27.], cd=cd)
mymap.header = header
# creation of the simulation
scan = PacsObservation.create_scan((header['CRVAL1'], header['CRVAL2']),
instrument_angle=0., length=60, nlegs=1, angle=20.)
simul = PacsSimulation(scan, 'red', policy_bad_detector='keep',
policy_other='keep')
# build the acquisition model
model = CompressionAverageOperator(simul.slice.compression_factor) * \
simul.get_projection_operator(header=header, npixels_per_sample=49)
# get the noiseless tod
tod = model(mymap)
filename = 'simul-'+str(uuid1())+'.fits'
try:
simul.save(filename, tod)
simul2 = PacsObservation(filename, policy_bad_detector='keep',
policy_other='keep')
status2 = simul2.status
tod2 = simul2.get_tod()
finally:
try:
os.remove(filename)
except:
pass
for field in simul.status.dtype.names:
if field == 'BAND': continue
assert_eq(simul.status[field], status2[field])
assert_eq(tod, tod2)
fields = [x for x in simul.slice.dtype.names if x not in ('filename','unit')]
for field in fields:
if getattr(simul.slice[0], field) != getattr(simul2.slice[0], field):
msg = "Field '" + field + "'"
if field == 'scan_step':
print(msg + ' not implemented.')
else:
assert False
def test_multiple_pointings():
ra = (23, 24)
dec = (50, 51)
instrument_angle=(10, 11)
scan_angle = (0, -90)
scan_length = (10, 20)
scan_nlegs = (2, 3)
scan_step = (147, 149)
scan_speed = (20, 60)
compression_factor = (4, 4)
acc = PacsSimulation.ACCELERATION
pointings = [PacsObservation.create_scan((r,d),sl,sst,None,ssp,acc,sn,sa,ia,
cf,False) for r,d,ia,sa,sl,sn,sst,ssp,cf in zip(ra,dec,
instrument_angle,scan_angle,scan_length,scan_nlegs,scan_step,
scan_speed,compression_factor)]
simul = PacsSimulation(pointings, 'blue')
s = simul.slice
assert len(s) == 2
assert_eq(s.ra, ra)
assert_eq(s.dec, dec)
assert_eq(s.instrument_angle, instrument_angle)
assert_eq(s.scan_angle, scan_angle)
assert_eq(s.scan_length, scan_length)
assert_eq(s.scan_nlegs, scan_nlegs)
assert_eq(s.scan_step, scan_step)
assert_eq(s.scan_speed, scan_speed)
assert_eq(s.compression_factor, compression_factor)
def func1(c, b):
pointing.time = time * c
simul = PacsSimulation(pointing, b)
assert simul.slice[0].mode == result[(c,b)]
assert simul.slice[0].compression_factor == c
def func3(c, b):
pointing.time = time * c
simul = PacsSimulation(pointing, b, mode='calibration')
assert simul.slice[0].compression_factor == c