code: def test_expand_dim_transform():
      di = np.array(range(10))
      eo = None
      fit_callback, transform_callback, fit_transform_callback = (
          TapeCallbackFunction(),
          TapeCallbackFunction(),
          TapeCallbackFunction(),
      )
      p = Pipeline(
          [
              ExpandDim(
                  HandleCallbackStep(
                      fit_callback,
                      transform_callback,
                      fit_transform_callback,
                  )
              )
          ]
      )
  
      outputs = p.transform(di)
  
      assert np.array_equal(outputs, di)
      assert fit_callback.data == []
      assert np.array_equal(
          np.array(transform_callback.data[0][0].di), np.array([di])
      )
      assert np.array_equal(
          np.array(transform_callback.data[0][0].eo), np.array([eo])
      )
      assert fit_transform_callback.data == []
  
file_path: Neuraxle/testing_neuraxle/steps/test_expand_dim.py
imports: from typing import List, Iterable
  import numpy
  from neuraxle.hyperparams.space import HyperparameterSamples
  from neuraxle.pipeline import Pipeline
  from neuraxle.steps.flow import ExpandDim
  from neuraxle.steps.misc import HandleCallbackStep, TapeCallbackFunction
label_desc: Use image node_id 1 for calling a global function with example usage: test_expand_dim_transform() without return types