code: def weighted_edges():
      # Four weighted edges originating from the center node and connected
      # to each corner
      edges_df = pd.DataFrame(
          {
              "id": np.arange(4),
              "source": np.zeros(4, dtype=np.int64),
              "target": np.arange(1, 5),
              "weight": np.ones(4),
          }
      )
      return edges_df.set_index("id")
  
file_path: datashader/datashader/tests/test_layout.py
imports: from __future__ import annotations
  import pytest
  import numpy
  import pandas
  from datashader.bundling import directly_connect_edges, hammer_bundle
  from datashader.layout import circular_layout, forceatlas2_layout, random_layout
label_desc: Use image node_id 4 for calling a global function with example usage: weighted_edges() and returns: edges_df