Fix: fix test

Author: jdeschamps

tests/lightning/dataset_ng/callbacks/prediction_writer_callbacks/test_write_tile_zarr_strategy.py

@@ -229,7 +229,7 @@ def test_update_data_shape(axes, data_shape, expected_shape):
         # simple usual shapes
         ("YX", (1, 1, 32, 64), (32, 64)),
         ("YX", (1, 1, 128, 32), (128, 32)),
-        ("ZYX", (1, 1, 32, 64, 64), (32, 64, 64)),
+        ("ZYX", (1, 1, 32, 64, 64), (1, 64, 64)),
         ("ZYX", (1, 1, 64, 128, 64), (1, 128, 64)),
         ("CYX", (1, 5, 64, 64), (1, 64, 64)),
         ("SYX", (5, 1, 64, 256), (1, 64, 128)),