Beispiel #1
0
 def test_pt(self):
     '''Test that pT is correctly set.'''
     cluster = Cluster(10., TVector3(1, 0, 0),
                       1)  #alice made this use default layer
     self.assertAlmostEqual(cluster.pt, 10.)
     cluster.set_energy(5.)
     self.assertAlmostEqual(cluster.pt, 5.)
Beispiel #2
0
 def test_acceptance(self):
     rootfile = TFile('test_cluster_acceptance.root', 'recreate')
     h_evseta = TH2F('h_evseta','cluster energy vs eta',
                     100, -5, 5, 100, 0, 15)
     h_ptvseta = TH2F('h_ptvseta','cluster pt vs eta',
                      100, -5, 5, 100, 0, 15)
     nclust = 1000.
     energies = np.random.uniform(0., 10., nclust)
     thetas = np.random.uniform(0, math.pi, nclust)
     costhetas = np.cos(thetas)
     sinthetas = np.sin(thetas)
     clusters = []
     for energy, cos, sin in zip(energies, costhetas, sinthetas):
         clusters.append(Cluster(energy, TVector3(sin,0,cos), 1, 1))
     ecal = cms.elements['ecal']
     smeared_clusters = []
     min_energy = -999.
     for cluster in clusters:
         smeared_cluster = simulator.smear_cluster(cluster, ecal)
         if smeared_cluster:
             h_evseta.Fill(smeared_cluster.position.Eta(),
                           smeared_cluster.energy)
             h_ptvseta.Fill(smeared_cluster.position.Eta(),
                            smeared_cluster.pt)
             smeared_clusters.append(smeared_cluster)
             if smeared_cluster.energy > min_energy:
                 min_energy = smeared_cluster.energy
     self.assertGreater(len(clusters), len(smeared_clusters))
     self.assertGreater(min_energy, ecal.emin)
     rootfile.Write()
     rootfile.Close()
 def test_smear(self):
     rootfile = TFile('test_cluster_smear.root', 'recreate')
     h_e = TH1F('h_e','cluster energy', 200, 5, 15.)
     energy = 10.
     cluster = Cluster(energy, TVector3(1,0,0), 1) #alice made this use default layer
     ecal = cms.elements['ecal']
     energies = []
     for i in range(10000):
         smeared = simulator.smear_cluster(cluster, ecal, accept=True)
         h_e.Fill(smeared.energy)
         energies.append(smeared.energy)
     npe = np.array(energies)
     mean = np.mean(npe)
     rms = np.std(npe)
     eres = ecal.energy_resolution(cluster.energy)
     self.assertAlmostEqual(mean, energy, places=1)
     self.assertAlmostEqual(rms, eres*energy, places=1)
     rootfile.Write()
     rootfile.Close()
Beispiel #4
0
 def test_acceptance(self):
     rootfile = TFile('test_cluster_acceptance.root', 'recreate')
     h_evseta = TH2F('h_evseta', 'cluster energy vs eta', 100, -5, 5, 100,
                     0, 15)
     h_ptvseta = TH2F('h_ptvseta', 'cluster pt vs eta', 100, -5, 5, 100, 0,
                      15)
     nclust = 1000
     # making 1000 deposits between 0 and 10 GeV
     energies = np.random.uniform(0., 10., nclust)
     # theta between 0 and pi
     thetas = np.random.uniform(0, math.pi, nclust)
     costhetas = np.cos(thetas)
     sinthetas = np.sin(thetas)
     clusters = []
     for energy, cos, sin in zip(energies, costhetas, sinthetas):
         clusters.append(Cluster(energy, TVector3(sin, 0, cos),
                                 1))  #alice made this use default layer
     ecal = cms.elements['ecal']
     smeared_clusters = []
     min_energy = -999.
     for cluster in clusters:
         smeared_cluster = simulator.make_and_store_smeared_cluster(
             cluster, ecal)
         if smeared_cluster:
             h_evseta.Fill(smeared_cluster.position.Eta(),
                           smeared_cluster.energy)
             h_ptvseta.Fill(smeared_cluster.position.Eta(),
                            smeared_cluster.pt)
             smeared_clusters.append(smeared_cluster)
             if smeared_cluster.energy > min_energy:
                 min_energy = smeared_cluster.energy
     # test that some clusters have been rejected
     # (not passing the acceptance)
     self.assertGreater(len(clusters), len(smeared_clusters))
     # test that the minimum cluster energy is larger than the
     # minimum ecal threshold
     ecal_min_thresh = min(ecal.emin.values())
     self.assertGreater(min_energy, ecal_min_thresh)
     rootfile.Write()
     rootfile.Close()
Beispiel #5
0
 def test_pt(self):
     '''Test that pT is correctly set.'''
     cluster = Cluster(10., TVector3(1,0,0), 1)  #alice made this use default layer
     self.assertAlmostEqual(cluster.pt, 10.)
     cluster.set_energy(5.)
     self.assertAlmostEqual(cluster.pt, 5.)
Beispiel #6
0
 def test_pt(self):
     '''Test that pT is correctly set.'''
     cluster = Cluster(10., TVector3(1,0,0), 1, 1)
     self.assertAlmostEqual(cluster.pt, 10.)
     cluster.set_energy(5.)
     self.assertAlmostEqual(cluster.pt, 5.)
Beispiel #7
0
 def test_pt(self):
     '''Test that pT is correctly set.'''
     cluster = Cluster(10., TVector3(1,0,0), 1, 1)
     self.assertAlmostEqual(cluster.pt, 10.)
     cluster.set_energy(5.)
     self.assertAlmostEqual(cluster.pt, 5.)