casa / casa5 / f72aa2708b7

Commits

Federico Montesino Pouzols authored f72aa2708b7 17 May 2018

fix/do rflag/clip/tfcrop preaverage tests, re CAS-11397

gcwrap/python/scripts/tests/test_flagdata.py

Modified

-3458
-3459
+        # Compare results
-3460
+        self.assertEqual(res1['flagged'], res2['flagged'])
-3461
-3462
+    def test_rflag_timeavg(self):
-3463
+        '''flagdata: rflag with time average and compare vs mstransform'''
-3464
-3465
+        # Unflag the original input data
-3466
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3467
--
+        # STEP 1: Time average with mstransform, then flagging with normal clip
-+
+        # STEP 1: Time average with mstransform, then flagging with normal rflag
-3469
+        mstransform(vis=self.vis,outputvis='test_rflag_timeavg_step1.ms',datacolumn='data',
-3470
+                    timeaverage=True,timebin='2s')
-3471
+        flagdata(vis='test_rflag_timeavg_step1.ms',flagbackup=False, mode='rflag',extendflags=False)
-3472
+        res1 = flagdata(vis='test_rflag_timeavg_step1.ms', mode='summary', spwchan=True)
-3473
-3474
+        # Unflag the original input data
-3475
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3476
--
+        # STEP 2: Flagging with clip using time average, then time average with mstransform
-+
+        # STEP 2: Flagging with rflag using time average, then time average with mstransform
-3478
+        flagdata(vis=self.vis, flagbackup=False, mode='rflag', datacolumn='DATA',
-3479
+                 timeavg=True, timebin='2s', extendflags=False)
-3480
+        mstransform(vis=self.vis,outputvis='test_rflag_timeavg_step2.ms',datacolumn='data',
-3481
+                    timeaverage=True,timebin='2s')
-3482
+        res2 = flagdata(vis='test_rflag_timeavg_step2.ms', mode='summary', spwchan=True)
-3483
-3484
+        # Compare results
--
+        self.assertEqual(res2['flagged'], 20)
--
-+
+        self.assertEqual(res2['flagged'], 20)
-+
-3487
+    def test_rflag_chanavg(self):
-3488
+        '''flagdata: rflag with chan average and compare vs mstransform'''
-3489
-3490
+        # Unflag the original input data
-3491
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3492
--
+        # STEP 1: Chan average with mstransform, then flagging with normal clip
-+
+        # STEP 1: Chan average with mstransform, then flagging with normal rflag
-3494
+        mstransform(vis=self.vis,outputvis='test_rflag_chanavg_step1.ms',datacolumn='data',
-3495
+                    chanaverage=True,chanbin=2)
-3496
+        flagdata(vis='test_rflag_chanavg_step1.ms',flagbackup=False, mode='rflag',extendflags=False)
-3497
+        res1 = flagdata(vis='test_rflag_chanavg_step1.ms', mode='summary', spwchan=True)
-3498
-3499
+        # Unflag the original input data
-3500
+        flagdata(vis=self.vis, flagbackup=False, mode='unflag')
-3501
--
+        # STEP 2: Flagging with clip using time average, then time average with mstransform
--
+        flagdata(vis=self.vis, flagbackup=False, mode='rflag', datacolumn='DATA',
-+
+        # STEP 2: Flagging with rflag using time average, then time average with mstransform
-+
+        flagdata(vis=self.vis, flagbackup=False, mode='rflag', datacolumn='DATA',
-3504
+                 channelavg=True, chanbin=2,extendflags=False)
-3505
+        mstransform(vis=self.vis, outputvis='test_rflag_chanavg_step2.ms',datacolumn='data',
-3506
+                    chanaverage=True,chanbin=2)
-3507
+        res2 = flagdata(vis='test_rflag_chanavg_step2.ms', mode='summary', spwchan=True)
-3508
-3509
+        # Compare results
-3510
+        self.assertEqual(res1['flagged'], res2['flagged'])
--
-+
-3512
+    def test_rflag_time_chanavg(self):
-3513
+        '''flagdata: rflag with time/chan average and compare vs mstransform'''
-3514
-3515
+        # Unflag the original input data
-3516
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3517
--
+        # STEP 1: Chan average with mstransform, then flagging with normal clip
-+
+        # STEP 1: Chan average with mstransform, then flagging with normal rflag
-3519
+        mstransform(vis=self.vis,outputvis='test_rflag_time_chanavg_step1.ms',datacolumn='data',
-3520
+                    timeaverage=True,timebin='2s',chanaverage=True,chanbin=2)
--
+        flagdata(vis='test_rflag_time_chanavg_step1.ms',flagbackup=False, mode='clip',extendflags=False)
-+
+        flagdata(vis='test_rflag_time_chanavg_step1.ms',flagbackup=False, mode='rflag',
-+
+                 extendflags=False)
-3523
+        res1 = flagdata(vis='test_rflag_time_chanavg_step1.ms', mode='summary', spwchan=True)
-3524
-3525
+        # Unflag the original input data
-3526
+        flagdata(vis=self.vis, flagbackup=False, mode='unflag')
-3527
--
+        # STEP 2: Flagging with clip using time average, then time average with mstransform
--
+        flagdata(vis=self.vis, flagbackup=False, mode='clip', datacolumn='DATA',
-+
+        # STEP 2: Flagging with rflag using time average, then time average with mstransform
-+
+        flagdata(vis=self.vis, flagbackup=False, mode='rflag', datacolumn='DATA',
-3530
+                 timeavg=True, timebin='2s', channelavg=True, chanbin=2, extendflags=False)
-3531
+        mstransform(vis=self.vis, outputvis='test_rflag_time_chanavg_step2.ms',datacolumn='data',
-3532
+                    timeaverage=True,timebin='2s',chanaverage=True,chanbin=2)
-3533
+        res2 = flagdata(vis='test_rflag_time_chanavg_step2.ms', mode='summary', spwchan=True)
-3534
--
+        # Compare results
--
+        self.assertEqual(res1['flagged'], res2['flagged'])
-+
+        # Check results
-+
+        self.assertEqual(res1['flagged'], 20)
-+
+        self.assertEqual(res2['flagged'], 62)
-3538
-3539
+    def test_tfcrop_timeavg(self):
-3540
+        '''flagdata: tfcrop with time average and compare vs mstransform'''
-3541
-3542
+        # Unflag the original input data
-3543
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3544
--
+        # STEP 1: Time average with mstransform, then flagging with normal clip
-+
+        # STEP 1: Time average with mstransform, then flagging with normal tfcrop
-3546
+        mstransform(vis=self.vis,outputvis='test_tfcrop_timeavg_step1.ms',datacolumn='data',
-3547
+                    timeaverage=True,timebin='2s')
--
+        flagdata(vis='test_tfcrop_timeavg_step1.ms',flagbackup=False, mode='rflag',extendflags=False)
-+
+        flagdata(vis='test_tfcrop_timeavg_step1.ms',flagbackup=False, mode='tfcrop',
-+
+                 extendflags=False)
-3550
+        res1 = flagdata(vis='test_tfcrop_timeavg_step1.ms', mode='summary', spwchan=True)
-3551
-3552
+        # Unflag the original input data
-3553
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3554
-3555
+        # STEP 2: Flagging with clip using time average, then time average with mstransform
--
+        flagdata(vis=self.vis, flagbackup=False, mode='rflag', datacolumn='DATA',
-+
+        flagdata(vis=self.vis, flagbackup=False, mode='tfcrop', datacolumn='DATA',
-3557
+                 timeavg=True, timebin='2s', extendflags=False)
-3558
+        mstransform(vis=self.vis,outputvis='test_tfcrop_timeavg_step2.ms',datacolumn='data',
-3559
+                    timeaverage=True,timebin='2s')
-3560
+        res2 = flagdata(vis='test_tfcrop_timeavg_step2.ms', mode='summary', spwchan=True)
-3561
--
+        # Compare results
--
+        self.assertEqual(res2['flagged'], 20)
-+
+        # Check results
-+
+        self.assertEqual(res1['flagged'], 0)
-+
+        self.assertEqual(res2['flagged'], 0)
-3565
-3566
+    def test_tfcrop_chanavg(self):
-3567
+        '''flagdata: tfcrop with chan average and compare vs mstransform'''
-3568
-3569
+        # Unflag the original input data
-3570
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3571
--
+        # STEP 1: Chan average with mstransform, then flagging with normal clip
-+
+        # STEP 1: Chan average with mstransform, then flagging with normal tfcrop
-3573
+        mstransform(vis=self.vis,outputvis='test_tfcrop_chanavg_step1.ms',datacolumn='data',
-3574
+                    chanaverage=True,chanbin=2)
--
+        flagdata(vis='test_tfcrop_chanavg_step1.ms',flagbackup=False, mode='rflag',extendflags=False)
-+
+        flagdata(vis='test_tfcrop_chanavg_step1.ms',flagbackup=False, mode='tfcrop',
-+
+                 extendflags=False)
-3577
+        res1 = flagdata(vis='test_tfcrop_chanavg_step1.ms', mode='summary', spwchan=True)
-3578
-3579
+        # Unflag the original input data
-3580
+        flagdata(vis=self.vis, flagbackup=False, mode='unflag')
-3581
--
+        # STEP 2: Flagging with clip using time average, then time average with mstransform
--
+        flagdata(vis=self.vis, flagbackup=False, mode='rflag', datacolumn='DATA',
-+
+        # STEP 2: Flagging with tfcrop using time average, then time average with mstransform
-+
+        flagdata(vis=self.vis, flagbackup=False, mode='tfcrop', datacolumn='DATA',
-3584
+                 channelavg=True, chanbin=2,extendflags=False)
-3585
+        mstransform(vis=self.vis, outputvis='test_tfcrop_chanavg_step2.ms',datacolumn='data',
-3586
+                    chanaverage=True,chanbin=2)
-3587
+        res2 = flagdata(vis='test_tfcrop_chanavg_step2.ms', mode='summary', spwchan=True)
-3588
-3589
+        # Compare results
-3590
+        self.assertEqual(res1['flagged'], res2['flagged'])
-3591
-3592
+    def test_tfcrop_time_chanavg(self):
-3593
+        '''flagdata: tfcrop with time/chan average and compare vs mstransform'''
-3594
-3595
+        # Unflag the original input data
-3596
+        flagdata(self.vis, flagbackup=False, mode='unflag')
-3597
--
+        # STEP 1: Chan average with mstransform, then flagging with normal clip
-+
+        # STEP 1: Chan average with mstransform, then flagging with normal tfcrop
-3599
+        mstransform(vis=self.vis,outputvis='test_tfcrop_time_chanavg_step1.ms',datacolumn='data',
-3600
+                    timeaverage=True,timebin='2s',chanaverage=True,chanbin=2)
--
+        flagdata(vis='test_tfcrop_time_chanavg_step1.ms',flagbackup=False, mode='clip',extendflags=False)
-+
+        flagdata(vis='test_tfcrop_time_chanavg_step1.ms',flagbackup=False, mode='tfcrop',extendflags=False)
-3602
+        res1 = flagdata(vis='test_tfcrop_time_chanavg_step1.ms', mode='summary', spwchan=True)
-3603
-3604
+        # Unflag the original input data
-3605
+        flagdata(vis=self.vis, flagbackup=False, mode='unflag')
-3606
--
+        # STEP 2: Flagging with clip using time average, then time average with mstransform
--
+        flagdata(vis=self.vis, flagbackup=False, mode='clip', datacolumn='DATA',
-+
+        # STEP 2: Flagging with tfcrop using time average, then time average with mstransform
-+
+        flagdata(vis=self.vis, flagbackup=False, mode='tfcrop', datacolumn='DATA',
-3609
+                 timeavg=True, timebin='2s', channelavg=True, chanbin=2, extendflags=False)
-3610
+        mstransform(vis=self.vis, outputvis='test_tfcrop_time_chanavg_step2.ms',datacolumn='data',
-3611
+                    timeaverage=True,timebin='2s',chanaverage=True,chanbin=2)
-3612
+        res2 = flagdata(vis='test_tfcrop_time_chanavg_step2.ms', mode='summary', spwchan=True)
-3613
-3614
+        # Compare results
-3615
+        self.assertEqual(res2['flagged'], res2['flagged'])
-3616
-3617
+class test_virtual_col(test_base):
-3618
+    def setUp(self):

casa5

Commits

Jira Issues

Add shortcut