Adapt StereoCombination test

cta-observatory · Jan 10, 2023 · ce1d8ad · ce1d8ad
1 parent 06e6420
commit ce1d8ad
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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,23 +118,23 @@ 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)
         },
         classification={
             "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)
         },
         classification={
             "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)
         },