code: def assert_equal(a, b):
      assert type(a) == type(b)
      if isinstance(a, pd.DataFrame):
          assert_frame_equal(a, b)
      elif isinstance(a, pd.Series):
          assert_series_equal(a, b)
      elif isinstance(a, pd.Index):
          assert_index_equal(a, b)
      else:
          assert a == b
  
file_path: distributed/distributed/tests/test_dask_collections.py
imports: from __future__ import annotations
  import pytest
  from packaging.version import parse
  from pandas.testing import assert_frame_equal, assert_index_equal, assert_series_equal
  import dask
  import dask.bag
  import dask.dataframe
  from distributed.client import wait
  from distributed.nanny import Nanny
  from distributed.utils_test import gen_cluster
label_desc: Use image node_id 1 for calling a global function with example usage: assert_equal(a, b) without return types