def test_setup_transit_adjust_params(self):
sherlock = Sherlock(False, None)
sherlock.setup_transit_adjust_params(max_runs=5,
period_protec=5,
period_min=1,
period_max=2,
bin_minutes=5,
run_cores=3,
snr_min=6,
sde_min=5,
fap_max=0.05,
mask_mode="subtract",
best_signal_algorithm="quorum",
quorum_strength=2)
self.assertEqual(5, sherlock.max_runs)
self.assertEqual(5, sherlock.period_protec)
self.assertEqual(1, sherlock.period_min)
self.assertEqual(2, sherlock.period_max)
self.assertEqual(5, sherlock.bin_minutes)
self.assertEqual(3, sherlock.run_cores)
self.assertEqual(6, sherlock.snr_min)
self.assertEqual(5, sherlock.sde_min)
self.assertEqual(0.05, sherlock.fap_max)
self.assertEqual("subtract", sherlock.mask_mode)
self.assertEqual("quorum", sherlock.best_signal_algorithm)
def test_scoring_algorithm(self):
sherlock = Sherlock(False, None)
sherlock.setup_transit_adjust_params(best_signal_algorithm="basic")
self.assertTrue(
isinstance(
sherlock.signal_score_selectors[
sherlock.best_signal_algorithm], BasicSignalSelector))
sherlock.setup_transit_adjust_params(
best_signal_algorithm="border-correct")
self.assertTrue(
isinstance(
sherlock.signal_score_selectors[
sherlock.best_signal_algorithm],
SnrBorderCorrectedSignalSelector))
sherlock.setup_transit_adjust_params(best_signal_algorithm="quorum")
self.assertTrue(
isinstance(
sherlock.signal_score_selectors[
sherlock.best_signal_algorithm],
QuorumSnrBorderCorrectedSignalSelector))
