Skip to content

Commit

Permalink
Fix breakage caused by reverting bad push
Browse files Browse the repository at this point in the history
  • Loading branch information
jeremykubica committed Dec 1, 2023
1 parent 2e93f07 commit 60c5caa
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tests/test_stats_filters.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
from kbmod.filters.stats_filters import *
from kbmod.result_list import *
from kbmod.search import *
from kbmod.trajectory_utils import make_trajectory


class test_basic_filters(unittest.TestCase):
Expand All @@ -12,10 +13,9 @@ def setUp(self):

self.rs = ResultList(self.times, track_filtered=True)
for i in range(10):
t = Trajectory()
row = ResultRow(t, self.num_times)
trj = make_trajectory(lh=float(i))
row = ResultRow(trj, self.num_times)
row.filter_indices([k for k in range(i)])
row.final_likelihood = float(i)
self.rs.append_result(row)

def test_filter_min_likelihood(self):
Expand Down Expand Up @@ -67,8 +67,8 @@ def test_filter_likelihood_mp(self):
# Create a lot more results.
rs = ResultList(self.times, track_filtered=True)
for i in range(1000):
row = ResultRow(Trajectory(), self.num_times)
row.final_likelihood = 0.01 * float(i)
trj = make_trajectory(lh=0.01 * float(i))
row = ResultRow(trj, self.num_times)
rs.append_result(row)
self.assertEqual(rs.num_results(), 1000)

Expand Down

0 comments on commit 60c5caa

Please sign in to comment.