casa5

#######################################################

#                                                                                                                         #

#           Regression/Test Script for Stokes-Selection in Imager and Clean                #

#                                                                                                                         #

#######################################################

#                                                                                                                         

#  (1) Given a 'true' [I,Q,U,V] vector, 

#       simulate MSs for [RR,RL,LR,LL], [RR,LL], [RR],[LL], [XX,XY,YX,YY], [XX,YY], [XX], [YY].

#  (2) Test all possible options, some valid, some not : ['I','Q','U','V', 'IV', 'QU', 'IQ', 'UV', 

#          'IQU', 'IUV', 'IQUV', 'RR', 'LL', 'RL','LR', 'RRLL','RLLR','XX','YY','XY','YX','XXYY','XYYX']

#  (3) For runs that produce an output image, the resulting output (peak pixel value) 

#       is compared with the 'true' [I,Q,U,V,RR,LL,XX,YY]. 

#  (4) Counts of 'pass' and 'fail' are, in the end, compared and used to detect regression.

#       Note : This script tests both valid and invalid options.

#

#                                                                                                                           

######################################################

#                                                                                                                            

# More tests that will appear here in the future 

# (See JIRA CAS 2587 : https://bugs.nrao.edu/browse/CAS-2587 ) :                            

#                                                                               

# (1) Tests with partially flagged data

# (2) Tests with different minor-cycle algorithms

#                                                                              

######################################################

​

import time

import os

​

# Init printing info.

import datetime

datestring=datetime.datetime.isoformat(datetime.datetime.today())

outfile='stokesimagingtest.'+datestring+'.log'

logfile=open(outfile,'w')

​

#####################################################

####  BEGIN : Helper function definition (tests are run after 'END')                 ####

#####################################################

​

################################

###  Simulate a small MS

################################

def makeMS(choice='RL_all',stokesvals=[1.0,1.0,0.0,0.0],parentpath='.'):

  dirname = parentpath+"/Point"

  choicelist = ['RL_all','RL_par','RR_only','LL_only','XY_all','XY_par','XX_only','YY_only'];

  if(not (choice in choicelist)):

    print >>logfile, 'Cannot find specs for ', choice;

    return;

  if(choice=='RL_all'):

    basename = dirname+"/point_linRL"

    msstokes='RR RL LR LL';

    feedtype='perfect R L';

  if(choice=='XY_all'):

    basename = dirname+"/point_linXY"

    msstokes='XX XY YX YY'; 

    feedtype='perfect X Y';

  if(choice=='RL_par'):

    basename = dirname+"/point_parRL"

    msstokes='RR LL';

    feedtype='perfect R L';

  if(choice=='XY_par'):

    basename = dirname+"/point_parXY"

    msstokes='XX YY';