code: def test_expand_dim_fit():
      handle_fit_callback = TapeCallbackFunction()
      handle_transform_callback = TapeCallbackFunction()
      handle_fit_transform_callback = TapeCallbackFunction()
      p = Pipeline(
          [
              ExpandDim(
                  HandleCallbackStep(
                      handle_fit_callback,
                      handle_transform_callback,
                      handle_fit_transform_callback,
                  )
              )
          ]
      )
  
      p = p.fit(np.array(range(10)), np.array(range(10)))
  
      assert handle_transform_callback.data == []
      assert handle_fit_transform_callback.data == []
      assert np.array_equal(
          np.array(handle_fit_callback.data[0][0].data_inputs),
          np.array([np.array(range(10))]),
      )
      assert np.array_equal(
          np.array(handle_fit_callback.data[0][0].expected_outputs),
          np.array([np.array(range(10))]),
      )
  
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 2 for calling a global function with example usage: test_expand_dim_fit() without return types