Skip to content

Commit f575c5b

Browse files
committed
fix calc_combs_min_distances unit tests
1 parent fd137bf commit f575c5b

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

src/ctapipe/reco/tests/test_telescope_event_handling.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,23 +75,26 @@ def test_calc_combs_min_distances():
7575
fov_lon_values = np.array([[1, 2], [3, 4], [5, 6]])
7676
fov_lat_values = np.array([[7, 8], [9, 10], [11, 12]])
7777
weights = np.array([1, 2, 3])
78+
dist_weights = np.array([[1, 1], [1, 1], [1, 0.55]], dtype=np.float64)
7879

7980
exp_comb_weights = np.array([3, 4, 5])
80-
exp_fov_lons = np.array([2.66666667, 4.25, 4.6])
81-
exp_fov_lats = np.array([8.66666667, 10.25, 10.6])
81+
exp_fov_lons = np.array([2.66666667, 5, 5.2])
82+
exp_fov_lats = np.array([8.66666667, 11, 11.2])
8283

8384
fov_lons_ev, fov_lats_ev, comb_weights_ev = calc_combs_min_distances_event(
8485
index_tel_combs,
8586
fov_lon_values,
8687
fov_lat_values,
8788
weights,
89+
dist_weights,
8890
)
8991

9092
fov_lons_tab, fov_lats_tab, comb_weights_tab = calc_combs_min_distances_table(
9193
index_tel_combs,
9294
fov_lon_values,
9395
fov_lat_values,
9496
weights,
97+
dist_weights,
9598
)
9699
assert np.allclose(fov_lons_ev, exp_fov_lons)
97100
assert np.allclose(fov_lons_tab, exp_fov_lons)

0 commit comments

Comments
 (0)