code: def assert_dict(da, db, assert_fn):
      assert len(da) == len(db)
      for k, v in da.items():
          assert k in db
          if not torch.is_tensor(v):
              continue
          u = db.get(k)
          assert_fn(u, v)
  
file_path: colossalai/tests/test_fx/test_tracer/test_hf_model/test_hf_diffuser.py
imports: import pytest
  import torch
  from colossalai.fx import symbolic_trace
  from colossalai.testing import clear_cache_before_run
  from colossalai.testing.random import seed_all
  from tests.kit.model_zoo import model_zoo
label_desc: Use image node_id 1 for calling a global function with example usage: assert_dict(da, db, assert_fn) without return types