Commits

ensure test_rflag doesn't leave a Four_ants MS with model removed, re CAS-11397
No tags

gcwrap/python/scripts/tests/test_flagdata.py

Modified
456 456 # Flags should be extended in time if > 50%, freq > 80% and
457 457 # will extend to the other polarizations too.
458 458 self.assertEqual(pos['spw']['0']['flagged'], pre['spw']['0']['flagged'])
459 459
460 460
461 461 class test_rflag(test_base):
462 462 """flagdata:: Test of mode = 'rflag'"""
463 463
464 464 def setUp(self):
465 465 self.setUp_data4tfcrop()
466 -
466 +
467 467 def test_rflag1(self):
468 468 '''flagdata:: mode = rflag : automatic thresholds'''
469 469 flagdata(vis=self.vis, mode='rflag', spw='9,10', timedev=[], freqdev=[], flagbackup=False,
470 470 extendflags=False)
471 471 res = flagdata(vis=self.vis, mode='summary',spw='7,9,10')
472 472 self.assertEqual(res['flagged'], 42728.0)
473 473 self.assertEqual(res['antenna']['ea19']['flagged'], 18411.0)
474 474 self.assertEqual(res['spw']['7']['flagged'], 0)
475 475
476 476 def test_rflag2(self):
647 647 fcmd = ["mode='summary' spw='7,9,10' name='InitFlags'",
648 648 "mode='rflag' spw='9,10' timedev=[] freqdev=[] extendflags=False",
649 649 "mode='summary' spw='7,9,10' name='RflagFlags'"]
650 650
651 651 res = flagdata(vis=self.vis, mode='list', inpfile=fcmd, flagbackup=False)
652 652 self.assertEqual(res['report0']['flagged'],0)
653 653 self.assertEqual(res['report1']['flagged'], 42728)
654 654 self.assertEqual(res['report1']['antenna']['ea19']['flagged'], 18411)
655 655 self.assertEqual(res['report1']['spw']['7']['flagged'], 0,)
656 656
657 - def test_rflag_residuals(self):
657 + def test_rflag_residual_data(self):
658 658 '''flagdata: rflag using MODEL and virtual MODEL columns'''
659 659 from tasks import delmod
660 660
661 661 # Delete model columns, if any
662 662 delmod(vis=self.vis,otf=True,scr=True)
663 663
664 664 # Create MODEL_COLUMN
665 665 setjy(vis=self.vis, field='3C286_A',usescratch=True)
666 666
667 667 # rflag
679 679 setjy(vis=self.vis, field='3C286_A',usescratch=False)
680 680
681 681 # rflag
682 682 flagdata(vis=self.vis, mode='rflag', spw='9,10',datacolumn='RESIDUAL_DATA',flagbackup=False)
683 683 # 444576.0 flags on virtual MODEL col
684 684 flags_vmod = flagdata(vis=self.vis, mode='summary',spw='9,10')
685 685
686 686 # Flags should be the same
687 687 self.assertTrue(flags_mod['flagged'],flags_vmod['flagged'])
688 688
689 + # This test is mischievous, manipulates the model column. Don't leave a messed up MS.
690 + os.system('rm -rf {0}'.format(self.vis))
689 691
690 692 class test_shadow(test_base):
691 693 def setUp(self):
692 694 self.setUp_shadowdata2()
693 695
694 696 def test_CAS2399(self):
695 697 '''flagdata: shadow by antennas not present in MS'''
696 698
697 699 if os.path.exists("cas2399.txt"):
698 700 os.system('rm -rf cas2399.txt')

Everything looks good. We'll let you know here if there's anything you should know about.

Add shortcut