Commits
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') |