code: def test_fit(data):
      a = sklearn.impute.SimpleImputer()
      b = dask_ml.impute.SimpleImputer()
  
      a.fit(X)
      b.fit(data)
  
      assert_estimator_equal(a, b, exclude=["statistics_"])
      np.testing.assert_array_almost_equal(
          a.statistics_, np.asarray(b.statistics_)
      )
  
file_path: dask-ml/tests/test_impute.py
imports: import dask.array
  import dask.dataframe
  import numpy
  import pandas
  import pandas.testing
  import pytest
  import sklearn.impute
  import dask_ml.datasets
  import dask_ml.impute
  from dask_ml._compat import DASK_2_26_0, PANDAS_1_2_0
  from dask_ml.utils import assert_estimator_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_fit(data) without return types