Tests fix.

luxonis · Sep 2, 2024 · f21a70e · f21a70e
1 parent 8499526
commit f21a70e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unittests/test_creators/test_depth.py b/tests/unittests/test_creators/test_depth.py
@@ -28,7 +28,7 @@ def test_not_3d_input():
 def test_wrong_input_shape():
     with pytest.raises(
         ValueError,
-        match=re.escape("Unexpected image shape. Expected CHW or HWC, got (3, 1, 3)."),
+        match=re.escape("Unexpected image shape. Expected NHW or HWN, got (3, 1, 3)."),
     ):
         create_depth_message(
             np.array([[[1, 2, 3]], [[4, 5, 6]], [[7, 8, 9]]]), "relative"