code: def test_relation_func(self):
      se_true = pd.DataFrame(
          {
              "larger_than": [False, True],
              "n_edges": [5, 2],
              "same_color": [3, 0],
          }
      )
      se_true.set_index("larger_than", inplace=True)
  
      se_test = g.partition_edges(
          relations="larger_than", relation_funcs={"same_color": "sum"}
      )
  
      assert_frame_equal(
          se_test.sort_index(axis=1), se_true.sort_index(axis=1)
      )
  
imports: import pytest
  import numpy
  import pandas
  from pandas.testing import assert_frame_equal
  from numpy.testing import assert_allclose
  from deepgraph import DeepGraph
label_desc: Use image node_id 3 for calling the TestPartitionEdges obj's underlying member method code with example usage: obj.test_relation_func() without return types