jean perrot / ctools

Commit 2ba7b22827502222838c9f9f6b66132cd5b1aec9

Authored by jean perrot
1 parent fa67c41c

refactoring test ctselect - add usetresh phase roi

Inline Side-by-side
Showing 1 changed file with 81 additions and 21 deletions
test/test_ctselect.py
  View file @2ba7b22
... ... @@ -68,8 +68,11 @@ class Test(test):
68 68 self.append(self._test_ignore_emin_value, 'CTSELECT - Test ignore emin value')
69 69 self.append(self._test_ROI_selection_displaced, 'CTSELECT - Test ROI selection displaced')
70 70 self.append(self._test_ROI_selection_outside, 'CTSELECT - Test ROI selection outside')
71   - self.append(self._test_set_ebounds_usetresh_user,'CTSELECT - Test set ebounds usetresh user'
72   - self.append(self._test_phase_cut, 'CTSELECT - Test phase cut')
  71 + self.append(self._test_set_ebounds_usetresh_user,'CTSELECT - Test set ebounds usetresh user')
  72 + self.append(self._test_set_ebounds_usetresh_default,'CTSELECT - Test set ebounds usetresh default')
  73 + self.append(self._test_phase_selection_2inter, 'CTSELECT - Test phase selection 2 interval')
  74 + self.append(self._test_phase_selection_first_inter, 'CTSELECT - Test phase selection 2 interval')
  75 + self.append(self._test_phase_selection_second_inter, 'CTSELECT - Test phase selection 2 interval')
73 76 self.append(self._test_invalid_phase_value_string, 'CTSELECT - Test invalid phase value or string')
74 77  
75 78 # Return
... ... @@ -345,13 +348,13 @@ class Test(test):
345 348 # Return
346 349 return
347 350  
348   - # Unit Test time selection function
  351 + # Unit Test time selection function --------------------------------------------
349 352 def _test_time_selection(self):
350 353 """
351 354 Unit Test time selection
352 355  
353 356 """
354   - select = self._setup(inobs=self._events,tmin=0.0,tmax=10.0,emin=0.2,emax=80,outobs='ctselect_pyTimeSelection.fits',logfile='ctselect_pyTimeSelection.log')
  357 + select = self._setup(inobs=self._events,tmin=0.0,tmax=10.0,emin=0.2,emax=80,outobs='ctselect_py_TimeSelection.fits',logfile='ctselect_py_TimeSelection.log')
355 358  
356 359 # Run ctselect tool
357 360 select.logFileOpen() # Make sure we get a log file
... ... @@ -359,7 +362,7 @@ class Test(test):
359 362 select.save()
360 363  
361 364 # Check result file
362   - self._check_result_file('ctselect_pyTimeSelection.fits', 'time selection',nevents=10)
  365 + self._check_result_file('ctselect_py_TimeSelection.fits', 'time selection',nevents=10)
363 366  
364 367  
365 368 self._teardown(select)
... ... @@ -367,13 +370,13 @@ class Test(test):
367 370 # Return
368 371 return
369 372  
370   - # Unit Test set ebounds function
  373 + # Unit Test set ebounds function -----------------------------------------------------
371 374 def _test_set_ebounds_usetresh_user(self):
372 375 """
373 376 Unit Test set ebounds with USER usetresh
374 377  
375 378 """
376   - select = self._setup(inobs=self._events,tmin=500.0,tmax=1000.0,emin=0.2,emax=80,usethres='USER',outobs='ctselect_pySetEboundsUser.fits',logfile='ctselect_pySetEboundsUser.log')
  379 + select = self._setup(inobs=self._events,tmin=500.0,tmax=1000.0,emin=0.2,emax=80,usethres='USER',outobs='ctselect_py_SetEboundsUser.fits',logfile='ctselect_py_SetEboundsUser.log')
377 380  
378 381 # Run ctselect tool
379 382 select.logFileOpen() # Make sure we get a log file
... ... @@ -381,7 +384,7 @@ class Test(test):
381 384 select.save()
382 385  
383 386 # Check result file
384   - self._check_result_file('ctselect_pySetEboundsUser.fits', ' USER usetresh')
  387 + self._check_result_file('ctselect_py_SetEboundsUser.fits', ' USER usetresh')
385 388  
386 389 self._teardown(select)
387 390  
... ... @@ -389,9 +392,27 @@ class Test(test):
389 392 return
390 393  
391 394  
  395 + def _test_set_ebounds_usetresh_default(self):
  396 + """
  397 + Unit Test set ebounds with DEFAULT usetresh
  398 +
  399 + """
  400 + select = self._setup(inobs=self._events,tmin=500.0,tmax=1000.0,emin=0.2,emax=80,usethres='DEFAULT',outobs='ctselect_py_SetEboundsDef.fits',logfile='ctselect_py_SetEboundsDef.log')
  401 +
  402 + # Run ctselect tool
  403 + select.logFileOpen() # Make sure we get a log file
  404 + select.run()
  405 + select.save()
  406 +
  407 + # Check result file
  408 + self._check_result_file('ctselect_py_SetEboundsDef.fits', ' DEFAULT usetresh',nevents=2000)
  409 +
  410 + self._teardown(select)
  411 +
  412 + # Return
  413 + return
392 414  
393 415  
394   -
395 416  
396 417 # Unit Test for ctselect copy
397 418 def _test_for_ctselect_copy(self):
... ... @@ -606,7 +627,7 @@ class Test(test):
606 627 # Return
607 628 return
608 629  
609   - # Unit Test ROI selection function
  630 + # Unit Test ROI selection function ----------------------------------------------------------------------------
610 631 def _test_ROI_selection_displaced(self):
611 632 """
612 633 Unit Test ROI selection
... ... @@ -615,14 +636,14 @@ class Test(test):
615 636 # Setup ctselect tool with an RoI that is displaced and of the same
616 637 # size as the original RoI. This should reduce the RoI radius
617 638 # automatically.
618   - select = self._setup(inobs=self._events,dec=24.01,rad=5,tmin=500.0,tmax=1000.0,emin=0.2,emax=80.0,outobs='ctselect_pyROIDisp.fits',logfile='ctselect_pyROIDisp.log',chatter=3)
  639 + select = self._setup(inobs=self._events,dec=24.01,rad=5,tmin=500.0,tmax=1000.0,emin=0.2,emax=80.0,outobs='ctselect_py_ROIDisp.fits',logfile='ctselect_py_ROIDisp.log',chatter=3)
619 640  
620 641 # Execute tool
621 642 select.logFileOpen() # Needed to get a new log file
622 643 select.execute()
623 644  
624 645 # Check result file
625   - self._check_result_file('ctselect_pyROIDisp.fits','ROI displaced same size', nevents=466, dec=24.01)
  646 + self._check_result_file('ctselect_py_ROIDisp.fits','ROI displaced same size', nevents=466, dec=24.01)
626 647  
627 648 self._teardown(select)
628 649  
... ... @@ -635,14 +656,14 @@ class Test(test):
635 656  
636 657 """
637 658  
638   - select = self._setup(inobs=self._events,dec=24.01,rad=5,tmin=500.0,tmax=1000.0,emin=0.2,emax=80.0,outobs='ctselect_pyROIOut.fits',logfile='ctselect_pyROIOut.log',chatter=3)
  659 + select = self._setup(inobs=self._events,dec=24.01,rad=5,tmin=500.0,tmax=1000.0,emin=0.2,emax=80.0,outobs='ctselect_py_ROIOut.fits',logfile='ctselect_py_ROIOut.log',chatter=3)
639 660  
640 661 # Now put the RoI outside the existing RoI. This should leave to
641 662 # an empty event list.
642 663 select['ra'] = 83.63
643 664 select['dec'] = -22.01
644   - select['outobs'] = 'ctselect_pyROIOut.fits[EVENTS;GTI]'
645   - select['logfile'] = 'ctselect_pyROIOut.log'
  665 + select['outobs'] = 'ctselect_py_ROIOut.fits[EVENTS;GTI]'
  666 + select['logfile'] = 'ctselect_py_ROIOut.log'
646 667 select['chatter'] = 3
647 668  
648 669 # Execute tool
... ... @@ -650,7 +671,7 @@ class Test(test):
650 671 select.execute()
651 672  
652 673 # Check result file
653   - self._check_result_file('ctselect_pyROIOut.fits','ROI outside existing ROI', nevents=0, dec=-22.01,
  674 + self._check_result_file('ctselect_py_ROIOut.fits','ROI outside existing ROI', nevents=0, dec=-22.01,
654 675 rad=5.0)
655 676  
656 677 self._teardown(select)
... ... @@ -659,26 +680,65 @@ class Test(test):
659 680 return
660 681  
661 682  
662   - # Unit Test phase cut
663   - def _test_phase_cut(self):
  683 + # Unit Test phase selection function ------------------------------------------------------------------------------------------------
  684 + def _test_phase_selection_2inter(self):
664 685 """
665   - Unit Test phase cut
  686 + Unit Test phase selction for a double interval
666 687  
667 688 """
668   - select = self._setup(inobs=self._phased_events,rad=5.0,tmin='INDEF',tmax='INDEF',phase='0.1:0.4,0.6:0.8',outobs='ctselect_py11.fits',logfile='ctselect_py11.log')
  689 + select = self._setup(inobs=self._phased_events,rad=5.0,tmin='INDEF',tmax='INDEF',phase='0.1:0.4,0.6:0.8',outobs='ctselect_py_Phase2Inter.fits',logfile='ctselect_py_Phase2Inter.log')
669 690  
670 691 # Run ctselect tool
671 692 select.logFileOpen() # Make sure we get a log file
672 693 select.execute()
673 694  
674 695 # Check result file
675   - self._check_result_file('ctselect_py11.fits','Phase cut', nevents=1909, rad=5.0)
  696 + self._check_result_file('ctselect_py_Phase2Inter.fits','Phase selection with 2 interval', nevents=1909, rad=5.0)
676 697  
677 698 self._teardown(select)
678 699  
679 700 # Return
680 701 return
681 702  
  703 + def _test_phase_selection_first_inter(self):
  704 + """
  705 + Unit Test phase selection for the first interval
  706 +
  707 + """
  708 + select = self._setup(inobs=self._phased_events,rad=5.0,tmin='INDEF',tmax='INDEF',phase='0.1:0.4',outobs='ctselect_py_PhaseFirstInter.fits',logfile='ctselect_py_PhaseFirstInter.log')
  709 +
  710 + # Run ctselect tool
  711 + select.logFileOpen() # Make sure we get a log file
  712 + select.execute()
  713 +
  714 + # Check result file
  715 + self._check_result_file('ctselect_py_PhaseFirstInter.fits','Phase selection with first interval', nevents=1170, rad=5.0)
  716 +
  717 + self._teardown(select)
  718 +
  719 + # Return
  720 + return
  721 +
  722 + def _test_phase_selection_second_inter(self):
  723 + """
  724 + Unit Test phase selection for the second interval
  725 +
  726 + """
  727 + select = self._setup(inobs=self._phased_events,rad=5.0,tmin='INDEF',tmax='INDEF',phase='0.6:0.8',outobs='ctselect_py_PhaseSecondInter.fits',logfile='ctselect_py_PhaseSecondInter.log')
  728 +
  729 + # Run ctselect tool
  730 + select.logFileOpen() # Make sure we get a log file
  731 + select.execute()
  732 +
  733 + # Check result file
  734 + self._check_result_file('ctselect_py_PhaseSecondInter.fits','Phase selection with second interval', nevents=739, rad=5.0)
  735 +
  736 + self._teardown(select)
  737 +
  738 + # Return
  739 + return
  740 +
  741 +
682 742 # Unit Test invalid phase value or string
683 743 def _test_invalid_phase_value_string(self):
684 744 """
... ...