code: def test_log_image_numpy_raises_exception_for_invalid_array_shape():
      import numpy as np
  
      with mlflow.start_run(), pytest.raises(
          ValueError, match="`image` must be a 2D or 3D array"
      ):
          mlflow.log_image(np.zeros((1,), dtype=np.uint8), "image.png")
  
file_path: mlflow/tests/tracking/test_log_image.py
imports: import os
  import posixpath
  from unittest import mock
  import pytest
  import mlflow
  from mlflow.utils.file_utils import local_file_uri_to_path
label_desc: Use image node_id 7 for calling a global function with example usage: test_log_image_numpy_raises_exception_for_invalid_array_shape() without return types