code: def test_mx_switch() -> None:
      a = (mx.nd.array([[1, 1, 0, 0]]), mx.nd.array([[1, 1, 1, 1]]))
      b = (mx.nd.array([[1, 0, 1, 0]]), mx.nd.array([[2, 2, 2, 2]]))
      c = mx.nd.array([[3, 3, 3, 3]])
      assert (
          (
              mx_switch(mx.nd, a, b, c)
              == mx.nd.array([1.0, 1.0, 2.0, 3.0])
          )
          .asnumpy()
          .all()
      )
  
file_path: gluonts/test/mx/test_mx_util.py
imports: import pytest
  import itertools
  import tempfile
  from pathlib import Path
  from typing import List
  import mxnet
  import numpy
  from gluonts.mx import Tensor
  from gluonts.mx.util import cumsum, export_symb_block, hybrid_block_to_symbol_block, import_symb_block, mx_switch, weighted_average
label_desc: Use image node_id 6 for calling a global function with example usage: test_mx_switch() without return types