code: def test_check_multi_class(setup):
      all_multi_class = ["auto", "multinomial", "ovr"]
      solver = "SGD"
  
      for multi_class in all_multi_class:
          checked_multi_class = _check_multi_class(
              multi_class, solver, 2
          )
          assert checked_multi_class == "multinomial"
  
      error_msg = re.escape(
          "Solver %s does not support "
          "an ovr backend with number of classes "
          "larger than 2." % solver
      )
      with pytest.raises(ValueError, match=error_msg):
          _check_multi_class("ovr", solver, 3)
  
      invalid_multi_class = "multiovr"
      error_msg = re.escape(
          "multi_class should be 'multinomial', "
          "'ovr' or 'auto'. Got %s." % invalid_multi_class
      )
      with pytest.raises(ValueError, match=error_msg):
          _check_multi_class(invalid_multi_class, solver, 3)
  
file_path: mars/mars/learn/glm/tests/test_logistic.py
imports: import re
  import pytest
  from sklearn.datasets import load_iris
  from .._logistic import _check_solver, _check_multi_class, LogisticRegression
label_desc: Use image node_id 2 for calling a global function with example usage: test_check_multi_class(setup) without return types