code: def test_aggregation(pd_es, dask_es):
      primitives = list_primitives()
      trans_primitives = []
      agg_list = primitives[primitives["type"] == "aggregation"][
          "name"
      ].tolist()
      agg_primitives = [
          prim for prim in agg_list if prim not in UNSUPPORTED
      ]
  
      assert pd_es == dask_es
  
      # Run DFS using each dataframe as a target and confirm results match
      for df in pd_es.dataframes:
          fm, _ = dfs(
              entityset=pd_es,
              target_dataframe_name=df.ww.name,
              trans_primitives=trans_primitives,
              agg_primitives=agg_primitives,
              cutoff_time=pd.Timestamp("2019-01-05 04:00"),
              max_depth=2,
          )
  
          dask_fm, _ = dfs(
              entityset=dask_es,
              target_dataframe_name=df.ww.name,
              trans_primitives=trans_primitives,
              agg_primitives=agg_primitives,
              cutoff_time=pd.Timestamp("2019-01-05 04:00"),
              max_depth=2,
          )
  
          # Categorical categories can be ordered differently, this makes sure they
          # are the same, including the index column
          index_col = df.ww.index
          fm = fm.reset_index()
          dask_fm = dask_fm.astype(fm.dtypes)
          fm = fm.set_index(index_col)
  
          pd.testing.assert_frame_equal(
              fm.sort_index(),
              to_pandas(dask_fm, index=index_col, sort_index=True),
          )
  
file_path: featuretools/featuretools/tests/primitive_tests/test_dask_primitives.py
imports: import pandas
  import pytest
  from featuretools import calculate_feature_matrix, dfs, list_primitives
  from featuretools.feature_base.cache import feature_cache
  from featuretools.primitives import get_aggregation_primitives, get_transform_primitives
  from featuretools.tests.testing_utils import to_pandas
  from featuretools.utils.gen_utils import Library
label_desc: Use image node_id 3 for calling a global function with example usage: test_aggregation(pd_es, dask_es) without return types