def test_conv_select(self):
self.assertRaises(ValueError, utils.conv_select, BCM_TEST, BPP_TEST,
[11], [11], wrong_dict, {})
conv = utils.conv_select(BCM_TEST, BPP_TEST, [11], [11],
conv_dict_formation['convergence_filter'], {})
self.assertTrue(np.all(conv.evol_type.isin([2, 4])))
self.assertTrue(np.all(conv.sep >= 0))
conv = utils.conv_select(BCM_TEST, BPP_TEST, [13, 14], range(0, 15),
conv_dict_1_SN['convergence_filter'], {})
self.assertEqual(len(conv), 0)
conv = utils.conv_select(BCM_TEST, BPP_TEST, [13, 14], range(0, 15),
conv_dict_2_SN['convergence_filter'], {})
self.assertEqual(len(conv), 0)
conv = utils.conv_select(BCM_TEST, BPP_TEST, [13, 14], range(0, 15),
conv_dict_disruption['convergence_filter'],
{})
self.assertEqual(len(conv), 4)
conv = utils.conv_select(BCM_TEST, BPP_TEST, [11], [11],
conv_dict_final_state['convergence_filter'],
{})
self.assertEqual(len(conv), int(len(BCM_TEST)))
conv = utils.conv_select(BCM_TEST, BPP_TEST, [13, 14], range(0, 15),
conv_dict_XRB_form['convergence_filter'], {})
self.assertEqual(len(conv), 0)
self.assertRaises(ValueError, utils.conv_select, BCM_TEST, BPP_TEST,
[11], [11], false_dict, {})
def test_conv_lims(self):
conv = utils.conv_select(BCM_TEST, BPP_TEST, [11], [11],
conv_dict_formation['convergence_filter'],
conv_lim_dict)
self.assertTrue(conv.sep.max() < conv_lim_dict["sep"][1])
self.assertTrue(conv.sep.min() > conv_lim_dict["sep"][0])