aboutsummaryrefslogtreecommitdiff
path: root/utils
diff options
context:
space:
mode:
authortlatorre <tlatorre@uchicago.edu>2020-12-08 09:46:19 -0600
committertlatorre <tlatorre@uchicago.edu>2020-12-08 09:46:19 -0600
commita22284c6fd6678990cb9d13a08d447d46fc9b8a8 (patch)
tree131246f69a3e376b53c97150dd02166891277dc9 /utils
parent5d8906508262d326450752204ac93119f355db88 (diff)
downloadsddm-a22284c6fd6678990cb9d13a08d447d46fc9b8a8.tar.gz
sddm-a22284c6fd6678990cb9d13a08d447d46fc9b8a8.tar.bz2
sddm-a22284c6fd6678990cb9d13a08d447d46fc9b8a8.zip
don't sample DM events when running --test
Diffstat (limited to 'utils')
-rwxr-xr-xutils/dm-search8
1 files changed, 1 insertions, 7 deletions
diff --git a/utils/dm-search b/utils/dm-search
index 5b9cbaa..d75fdcc 100755
--- a/utils/dm-search
+++ b/utils/dm-search
@@ -667,21 +667,15 @@ if __name__ == '__main__':
N_atm = len(data_atm_mc)*xtrue[0]/atmo_scale_factor
N_muon = len(muon)*xtrue[5]/muon_scale_factor
N_muon_atm = len(muon_atm)*xtrue[5]/muon_scale_factor
- N_dm = xtrue[6]
# Calculate observed number of events
n = np.random.poisson(N)
n_atm = np.random.poisson(N_atm)
n_muon = np.random.poisson(N_muon)
n_muon_atm = np.random.poisson(N_muon_atm)
- n_dm = np.random.poisson(N_dm)
-
- dm_particle_id = np.random.choice([2020,2222])
- dm_mass = np.random.uniform(20,10e3)
- dm_energy = dm_mass
# Sample data from Monte Carlo
- data = pd.concat((data_mc.sample(n=n,replace=True), muon.sample(n=n_muon,replace=True),get_dm_sample(n_dm,dm_particle_id,dm_mass,dm_energy)))
+ data = pd.concat((data_mc.sample(n=n,replace=True), muon.sample(n=n_muon,replace=True)))
data_atm = pd.concat((data_atm_mc.sample(n=n_atm,replace=True), muon_atm.sample(n=n_muon_atm,replace=True)))
# Smear the energies by the additional energy resolution