def test_is_track_at_tpc_border_for_MC(self):
     self.load_selector()
     self.sel.data_type = 'MC'
     track = Mock()
     track.qMC=3; track.ptMC = 2; track.phiMC = 0.12 - pi/18
     self.assertTrue(self.sel.is_track_at_tpc_border(track))
     track.qMC = -3; track.phiMC=.23 - pi/18
     self.assertTrue(self.sel.is_track_at_tpc_border(track))
     track.qMC = 3; track.phiMC = 0.3  - pi/18
     self.assertFalse(self.sel.is_track_at_tpc_border(track))
    def test__validated_mc_track(self):
        """Test the mc track validator for various scenarios"""
        self.load_selector()
        self.sel.data_type = 'MC'
        track = Mock()
        track.etaMC = 0  # dont care about eta here

        # only pos charge, no TPC border treatment!
        self.sel.only_charge = 3; self.sel.exclude_tpc_border = False
        track.qMC=3; track.ptMC = 2; track.phiMC = 0.12 - pi/18        
        self.assertTrue(self.sel._validate_mc_track(track))
        track.qMC=-3
        self.assertFalse(self.sel._validate_mc_track(track))

        # only pos charge, with TPC border exclusion
        self.sel.only_charge = 3; self.sel.exclude_tpc_border = True
        track.qMC=3; track.ptMC = 2; track.phiMC = 0.12 - pi/18        
        self.assertFalse(self.sel._validate_mc_track(track))
        track.qMC = 3; track.phiMC = 0.3  - pi/18
        self.assertTrue(self.sel._validate_mc_track(track))

        # exclude TPC borders, include both charges
        self.sel.only_charge = None; self.sel.exclude_tpc_border = True
        track.qMC = 3; track.ptMC = 2; track.phiMC = 0.12 - pi/18        
        self.assertFalse(self.sel._validate_mc_track(track))
        track.qMC = -3; track.phiMC = 0.05  - pi/18
        self.assertTrue(self.sel._validate_mc_track(track))
 def load_hard_event(self, option='valid'):
     """fChain needs extra care, calling function needs to be decorated with
     @patch.object(MyPySelector, 'fChain', Mock())"""
     e = self.sel.fChain.event
     e.trig = 1; e.vtxstatus = 1; e.zvtxMC=0; e.ptmaxMC = 1 + self.pt_threshold
     e.cent = 50; e.zvtx=0; e.ptmax = 1 + self.pt_threshold;
     t = Mock()
     t.qMC=3; t.ptMC = 1.5; t.phiMC = 0.12 - pi/18; t.etaMC = 0.0
     t.q=3; t.pt = 1.5; t.phi = 0.12 - pi/18; t.eta = 0; t.filter=1; t.primary=1;
     t.dcaxy=0.0; t.dcaz = 0.0
     self.sel.get_tracks_generator = Mock(return_value=[t,])
     if option == 'valid':
         return None
     if option == 'bad threshold':
         e.ptmaxMC = e.ptmax = self.pt_threshold * .5
         return None