code: def test_wrong_column_name_number():
      with pytest.raises(AssertionError) as excinfo:
          heatmap(
              np.random.random((10, 5)), column_names=["a", "b", "c"]
          )
          assert excinfo.value.message == (
              "len(column_names) (got 3)"
              " should be equal to number of"
              " rows in the input "
              " array (expect 5)."
          )
  
imports: import matplotlib.pyplot
  import numpy
  import pytest
  from mlxtend.plotting import heatmap
label_desc: Use image node_id 2 for calling a global function with example usage: test_wrong_column_name_number() without return types