code: def test_dropout_remover(self):  # type: () -> None
      inputs = [("input", (1, 3, 50, 50))]
      outputs = [("out", (1, 5, 50, 50), TensorProto.FLOAT)]
      weight = numpy_helper.from_array(
          _random_array((5, 3, 1, 1)), name="weight"
      )
      conv = helper.make_node(
          "Conv",
          inputs=["input", "weight"],
          outputs=["conv_output"],
          kernel_shape=(1, 1),
          strides=(1, 1),
      )
      drop = helper.make_node(
          "Dropout",
          inputs=["conv_output"],
          outputs=["drop_output"],
      )
      exp = helper.make_node(
          "Exp", inputs=["drop_output"], outputs=["out"]
      )
  
      onnx_model = _onnx_create_model(
          [conv, drop, exp], inputs, outputs
      )
  
      graph = Graph.from_onnx(onnx_model.graph, onnx_ir_version=5)
      new_graph = graph.transformed([DropoutRemover()])
      self.assertEqual(len(graph.nodes), 3)
      self.assertEqual(len(new_graph.nodes), 2)
      self.assertEqual(new_graph.nodes[0].inputs[0], "input")
      self.assertEqual(
          new_graph.nodes[1].inputs[0], new_graph.nodes[0].outputs[0]
      )
      self.assertEqual(new_graph.nodes[1].outputs[0], "out")
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from __future__ import unicode_literals
  import pytest
  import unittest
  import numpy
  import numpy.testing
  from coremltools._deps import _HAS_ONNX, MSG_ONNX_NOT_FOUND
label_desc: Use image node_id 1 for calling the NodeRemoverTests obj's underlying member method code with example usage: obj.test_dropout_remover() without return types