code: def test_multi_class():
      estimator = xgb.XGBClassifier(
          n_estimators=1, random_state=1, max_depth=1
      )
      estimator.fit(np.array([[1], [2], [3]]), np.array([0, 1, 2]))
  
      assembler = XGBoostModelAssemblerSelector(estimator)
      actual = assembler.assemble()
  
      num_expr = ast.BinNumExpr(
          ast.NumVal(0.5),
          ast.NumVal(-7.663454759665456e-09),
          ast.BinNumOpType.ADD,
      )
      expected = ast.SoftmaxExpr([num_expr] * 3)
  
      assert utils.cmp_exprs(actual, expected)
  
file_path: m2cgen/tests/assemblers/test_boosting_xgboost.py
imports: import numpy
  import xgboost
  from m2cgen import ast
  from m2cgen.assemblers import XGBoostModelAssemblerSelector
  from m2cgen.assemblers.utils import apply_op_to_expressions
  from tests import utils
label_desc: Use image node_id 2 for calling a global function with example usage: test_multi_class() without return types