def test_fitsheader(): def func(n, hdr): hdr_local = scatter_fitsheader(hdr) assert_equal(hdr_local['NAXIS2'], distribute(n)) assert gather_fitsheader(hdr_local) == hdr for n in range(10): yield func, n, create_fitsheader((5,n), crval=(1,1), cdelt=1.)
def test_fitsheader(): def func(n, hdr): hdr_local = scatter_fitsheader(hdr) assert_equal(hdr_local['NAXIS2'], distribute(n)) assert gather_fitsheader(hdr_local) == hdr for n in range(10): yield func, n, create_fitsheader((5, n), crval=(1, 1), cdelt=1.)
#solver = scipy.sparse.linalg.bicgstab solver = tamasis.solvers.cg tol = 1.e-2 mtol = 1.e-10 maxiter = 100 hyper = 1 rank = MPI.COMM_WORLD.rank tamasis.var.verbose = True profile = None#'test_rls.png' data_dir = os.path.dirname(__file__) + '/data/' # reference map (no communication) header_ref_global = create_fitsheader((97,108), cdelt=3.2/3600, crval=(245.998427916727,61.5147650744551)) comm_tod = MPI.COMM_SELF comm_map = MPI.COMM_SELF obs_ref = PacsObservation(data_dir + 'frames_blue.fits', comm=comm_tod) obs_ref.pointing.chop = 0 tod_ref = obs_ref.get_tod() model_ref = MaskOperator(tod_ref.mask) * obs_ref.get_projection_operator( downsampling=True, npixels_per_sample=6, commin=comm_map, header=header_ref_global) map_ref_global = mapper_rls(tod_ref, model_ref, tol=tol, maxiter=maxiter, solver=solver, hyper=hyper) header_ref_global = map_ref_global.header cov_ref_global = map_ref_global.coverage mask_ref_global = map_ref_global.coverage == 0 tolocal = MPIDistributionGlobalOperator(map_ref_global.shape, attrin={'header':header_ref_global})
from tamasis.utils import assert_all_eq #solver = scipy.sparse.linalg.bicgstab solver = tamasis.solvers.cg tol = 1.e-2 mtol = 1.e-10 maxiter = 100 hyper = 1 rank = MPI.COMM_WORLD.rank tamasis.var.verbose = True profile = None #'test_rls.png' data_dir = os.path.dirname(__file__) + '/data/' # reference map (no communication) header_ref_global = create_fitsheader( (97, 108), cdelt=3.2 / 3600, crval=(245.998427916727, 61.5147650744551)) comm_tod = MPI.COMM_SELF comm_map = MPI.COMM_SELF obs_ref = PacsObservation(data_dir + 'frames_blue.fits', comm=comm_tod) obs_ref.pointing.chop = 0 tod_ref = obs_ref.get_tod() model_ref = MaskOperator(tod_ref.mask) * obs_ref.get_projection_operator( downsampling=True, npixels_per_sample=6, commin=comm_map, header=header_ref_global) map_ref_global = mapper_rls(tod_ref, model_ref, tol=tol, maxiter=maxiter, solver=solver,