code: def test_class_names_must_match(
      target_analysis_binary, ploomber_value_error_message
  ):
      """
      Assert error raised when more classes are in data than specified
      """
  
      _, _, y_train, y_test = target_analysis_binary
  
      with pytest.raises(
          ValueError, match=ploomber_value_error_message
      ):
          plot.target_analysis(
              y_train, y_test=y_test, labels=["a", "b", "c"]
          )
  
file_path: sklearn-evaluation/tests/test_target_analysis.py
imports: import pytest
  import numpy
  import pandas
  from sklearn import datasets
  from matplotlib.testing.decorators import image_comparison
  from sklearn.model_selection import train_test_split
  from sklearn_evaluation import plot
label_desc: Use image node_id 2 for calling a global function with example usage: test_class_names_must_match(target_analysis_binary, ploomber_value_error_message) without return types