diff --git a/ctapipe/reco/tests/test_stereo_combination.py b/ctapipe/reco/tests/test_stereo_combination.py index 2cd8aec4189..bea2f6f3938 100644 --- a/ctapipe/reco/tests/test_stereo_combination.py +++ b/ctapipe/reco/tests/test_stereo_combination.py @@ -5,11 +5,11 @@ from numpy.testing import assert_array_equal from ctapipe.containers import ( + ArrayDL2Container, ArrayEventContainer, HillasParametersContainer, ImageParametersContainer, ParticleClassificationContainer, - ReconstructedContainer, ReconstructedEnergyContainer, ) from ctapipe.reco.stereo_combination import StereoMeanCombiner @@ -118,7 +118,7 @@ def test_mean_prediction_single_event(weights): ) ) - event.dl2.tel[25] = ReconstructedContainer( + event.dl2.tel[25] = ArrayDL2Container( energy={ "dummy": ReconstructedEnergyContainer(energy=10 * u.GeV, is_valid=True) }, @@ -126,7 +126,7 @@ def test_mean_prediction_single_event(weights): "dummy": ParticleClassificationContainer(prediction=1.0, is_valid=True) }, ) - event.dl2.tel[125] = ReconstructedContainer( + event.dl2.tel[125] = ArrayDL2Container( energy={ "dummy": ReconstructedEnergyContainer(energy=20 * u.GeV, is_valid=True) }, @@ -134,7 +134,7 @@ def test_mean_prediction_single_event(weights): "dummy": ParticleClassificationContainer(prediction=0.0, is_valid=True) }, ) - event.dl2.tel[130] = ReconstructedContainer( + event.dl2.tel[130] = ArrayDL2Container( energy={ "dummy": ReconstructedEnergyContainer(energy=0.04 * u.TeV, is_valid=True) },