code: def test_multi_class():
      estimator = linear_model.LogisticRegression()
      estimator.coef_ = np.array([[1, 2], [3, 4], [5, 6]])
      estimator.intercept_ = np.array([7, 8, 9])
  
      assembler = assemblers.SklearnLinearModelAssembler(estimator)
      actual = assembler.assemble()
  
      expected = ast.VectorVal(
          [
              ast.BinNumExpr(
                  ast.BinNumExpr(
                      ast.NumVal(7),
                      ast.BinNumExpr(
                          ast.FeatureRef(0),
                          ast.NumVal(1),
                          ast.BinNumOpType.MUL,
                      ),
                      ast.BinNumOpType.ADD,
                  ),
                  ast.BinNumExpr(
                      ast.FeatureRef(1),
                      ast.NumVal(2),
                      ast.BinNumOpType.MUL,
                  ),
                  ast.BinNumOpType.ADD,
              ),
              ast.BinNumExpr(
                  ast.BinNumExpr(
                      ast.NumVal(8),
                      ast.BinNumExpr(
                          ast.FeatureRef(0),
                          ast.NumVal(3),
                          ast.BinNumOpType.MUL,
                      ),
                      ast.BinNumOpType.ADD,
                  ),
                  ast.BinNumExpr(
                      ast.FeatureRef(1),
                      ast.NumVal(4),
                      ast.BinNumOpType.MUL,
                  ),
                  ast.BinNumOpType.ADD,
              ),
              ast.BinNumExpr(
                  ast.BinNumExpr(
                      ast.NumVal(9),
                      ast.BinNumExpr(
                          ast.FeatureRef(0),
                          ast.NumVal(5),
                          ast.BinNumOpType.MUL,
                      ),
                      ast.BinNumOpType.ADD,
                  ),
                  ast.BinNumExpr(
                      ast.FeatureRef(1),
                      ast.NumVal(6),
                      ast.BinNumOpType.MUL,
                  ),
                  ast.BinNumOpType.ADD,
              ),
          ]
      )
  
      assert cmp_exprs(actual, expected)
  
file_path: m2cgen/tests/assemblers/test_linear_sklearn.py
imports: import numpy
  import pytest
  from sklearn import linear_model
  from m2cgen import assemblers, ast
  from tests.utils import cmp_exprs
label_desc: Use image node_id 3 for calling a global function with example usage: test_multi_class() without return types