code: def test_api_wrapper_works():
      test_client = setup_api()
      data_inputs = [
          [0, 1, 2],
          [3, 4, 5],
      ]
  
      json_response = test_client.get("/", json=data_inputs)
  
      predictions_np_arr = np.array(json_response.json["predictions"])
      expected_outputs = 2 * np.array(data_inputs)
      assert np.array_equal(predictions_np_arr, expected_outputs)
  
imports: import numpy
  from neuraxle.rest.flask import JSONDataResponseEncoder, JSONDataBodyDecoder, FlaskRestApiWrapper
  from neuraxle.base import BaseTransformer
label_desc: Use image node_id 2 for calling a global function with example usage: test_api_wrapper_works() without return types