use bool instead of int

Co-authored-by: Régis Terrier <regis.terrier@m4x.org>
gammapy · Jun 20, 2023 · 8027c76 · 8027c76
1 parent 8ce693b
commit 8027c76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gammapy/datasets/tests/test_map.py b/gammapy/datasets/tests/test_map.py
@@ -1914,7 +1914,7 @@ def test_create_psf_reco(geom):
 def test_to_masked():
     axis = MapAxis.from_energy_bounds(1, 10, 2, unit="TeV")
     geom = WcsGeom.create(npix=(10, 10), binsz=0.05, axes=[axis])
-    counts = Map.from_geom(geom, data=np.ones((2, 10, 10)))
+    counts = Map.from_geom(geom, data=1)
     mask_data = np.ones((2, 10, 10), dtype=bool)
     mask_data[0][5:8] = 0
     mask = Map.from_geom(geom, data=mask_data, dtype=bool)