Skip to content
Snippets Groups Projects
Commit 2c314f3f authored by Colm Talbot's avatar Colm Talbot
Browse files

add test

parent 11b9ac78
No related branches found
No related tags found
1 merge request!222Resolve "BNS: ability to sample in and convert to \tilde{Lambda} and \delta\tilde{\Lambda}"
Pipeline #33197 passed
......@@ -97,8 +97,20 @@ class TestBasicConversions(unittest.TestCase):
self.assertTrue(all([abs(self.lambda_1 - lambda_1) < 1e-5,
abs(self.lambda_2 - lambda_2) < 1e-5]))
def test_lambda_1_lambda_2_to_lambda_tilde(self):
lambda_tilde = \
conversion.lambda_1_lambda_2_to_lambda_tilde(
self.lambda_1, self.lambda_2, self.mass_1, self.mass_2)
self.assertTrue((self.lambda_tilde - lambda_tilde) < 1e-5)
class TestConvertToLALBBHParams(unittest.TestCase):
def test_lambda_1_lambda_2_to_delta_lambda(self):
delta_lambda = \
conversion.lambda_1_lambda_2_to_lambda_tilde(
self.lambda_1, self.lambda_2, self.mass_1, self.mass_2)
self.assertTrue((self.delta_lambda - delta_lambda) < 1e-5)
class TestConvertToLALParams(unittest.TestCase):
def setUp(self):
self.search_keys = []
......@@ -110,7 +122,7 @@ class TestConvertToLALBBHParams(unittest.TestCase):
del self.parameters
del self.remove
def test_cos_angle_to_angle_conversion(self):
def test_bbh_cos_angle_to_angle_conversion(self):
with mock.patch('numpy.arccos') as m:
m.return_value = 42
self.parameters['cos_tilt_1'] = 1
......@@ -119,6 +131,16 @@ class TestConvertToLALBBHParams(unittest.TestCase):
self.parameters)
self.assertDictEqual(self.parameters, dict(tilt_1=42, cos_tilt_1=1))
def test_bns_cos_angle_to_angle_conversion(self):
with mock.patch('numpy.arccos') as m:
m.return_value = 42
self.parameters['cos_tilt_1'] = 1
self.parameters, _ = \
conversion.convert_to_lal_binary_neutron_star_parameters(
self.parameters)
self.assertDictEqual(self.parameters, dict(
tilt_1=42, cos_tilt_1=1, lambda_1=0, lambda_2=0))
if __name__ == '__main__':
unittest.main()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment