updated gstlal-ugly/python/injection_utils.py

gstlal-ugly/python/injection_utils.py

@@ -132,7 +132,7 @@ def draw_mass_pair_uniform(mass_dict, alpha=-2.35, beta=2.):
     '''
     min_mass = mass_dict['min_mass']
     max_mass = mass_dict['max_mass']
-    assert min_mass > 0
+    assert min_mass is not None
     assert max_mass >= min_mass
 
     while True:
@@ -151,8 +151,8 @@ def draw_mass_distinct_uniform(mass_dict):
     max_mass1 = mass_dict['max_mass1']
     min_mass2 = mass_dict['min_mass2']
     max_mass2 = mass_dict['max_mass2']
-    assert min_mass1 > 0
-    assert min_mass2 > 0
+    assert min_mass1 is not None
+    assert min_mass2 is not None
     assert max_mass1 >= min_mass1
     assert max_mass2 >= min_mass2
     assert mass_dict['max_mtotal'] > min_mass1 + min_mass2
@@ -215,8 +215,8 @@ def draw_mass_pair_normal(mass_dict):
 
     mean_mass = mass_dict['mean_mass']
     sigma_mass = mass_dict['sigma_mass']
-    assert mean_mass > 0
-    assert sigma_mass > 0
+    assert mean_mass is not None
+    assert sigma_mass is not None
     assert mass_dict['min_mass'] > 0
     assert mass_dict['max_mass'] > mass_dict['min_mass']
 
@@ -239,7 +239,7 @@ def draw_mass_distinct_uniform_uniforminlog(mass_dict):
     max_mass1 = mass_dict['max_mass1']
     lnm2min = numpy.log(mass_dict['min_mass2'])
     lnm2max = numpy.log(mass_dict['max_mass2'])
-    assert min_mass1 > 0
+    assert min_mass1 is not None
     assert max_mass1 >= min_mass1
     assert lnm2max >= lnm2min
 
@@ -263,11 +263,11 @@ def draw_mass_distinct_normal_imf(mass_dict, alpha_salpeter=-2.35):
     max_mass1 = mass_dict['max_mass1']
     min_mass2 = mass_dict['min_mass2']
     max_mass2 = mass_dict['max_mass2']
-    assert mean_mass > 0
-    assert sigma_mass > 0
-    assert min_mass1 > 0
+    assert mean_mass is not None
+    assert sigma_mass is not None
+    assert min_mass1 is not None
     assert max_mass1 >= min_mass1
-    assert min_mass2 > 0
+    assert min_mass2 is not None
     assert max_mass2 >= min_mass2
     assert mass_dict['max_mtotal'] > min_mass1 + min_mass2
 
@@ -294,9 +294,9 @@ def draw_mass_distinct_uniform_imf(mass_dict, alpha_salpeter=-2.35):
     max_mass1 = mass_dict['max_mass1']
     min_mass2 = mass_dict['min_mass2']
     max_mass2 = mass_dict['max_mass2']
-    assert min_mass1 > 0
+    assert min_mass1 is not None
     assert max_mass1 >= min_mass1
-    assert min_mass2 > 0
+    assert min_mass2 is not None
     assert max_mass2 >= min_mass2
 
     a, b = powerlaw_setup(min_mass2, max_mass2, alpha_salpeter)