code: def test_plot_pi_n_cols(exp, n_cols):
      """Test if the number of figure columns matches the expected one."""
      ax = plot_permutation_importance(exp=exp, n_cols=n_cols)
      assert ax.shape[-1] == n_cols
  
imports: import re
  from copy import deepcopy
  import matplotlib.pyplot
  import numpy
  import pytest
  from pytest_lazyfixture import lazy_fixture
  from sklearn.metrics import accuracy_score, f1_score, max_error
  from alibi.api.defaults import DEFAULT_DATA_PERMUTATION_IMPORTANCE, DEFAULT_META_PERMUTATION_IMPORTANCE
  from alibi.api.interfaces import Explanation
  from alibi.explainers import PermutationImportance, plot_permutation_importance
label_desc: Use image node_id 23 for calling a global function with example usage: test_plot_pi_n_cols(exp, n_cols) without return types