Refactor and parametrize 'compute_distance2continuum'.

tardis/montecarlo/tests/test_cmontecarlo.py

@@ -138,11 +138,22 @@ def test_compute_distance2line(packet_params, expected_stat, expected_dist, pack
     assert obtained_stat == expected_stat
 
 
-def test_compute_distance2continuum():
-    distance_to_electron = 4.359272608766106e+28
-    tests.test_compute_distance2continuum.restype = c_double
-    np.testing.assert_almost_equal(tests.test_compute_distance2continuum(),
-                                  distance_to_electron)
+@pytest.mark.parametrize(
+    ['packet_params', 'model_params', 'expected_params'],
+    [({'virtual_packet': 0}, {'cont_status': 0}, {'chi_cont': 6.652486e-16, 'd_cont': 4.359272608766106e+28}),
+     ({'virtual_packet': 0}, {'cont_status': 1}, {'chi_cont': 6.652486e-16, 'd_cont': 4.358643523052283e+28}),
+     ({'virtual_packet': 1}, {'cont_status': 0}, {'chi_cont': 6.652486e-16, 'd_cont': 1e+99}),
+     ({'virtual_packet': 1}, {'cont_status': 1}, {'chi_cont': 6.652486e-16, 'd_cont': 1e+99})]
+)
+def test_compute_distance2continuum(packet_params, model_params, expected_params, packet, model):
+    packet.virtual_packet = packet_params['virtual_packet']
+    model.cont_status = model_params['cont_status']
+
+    cmontecarlo_methods.compute_distance2continuum.restype = c_voidp
+    cmontecarlo_methods.compute_distance2continuum(byref(packet), byref(model))
+
+    np.testing.assert_almost_equal(packet.chi_cont, expected_params['chi_cont'])
+    np.testing.assert_almost_equal(packet.d_cont, expected_params['d_cont'])
 
 
 def test_rpacket_doppler_factor():