code: def test_relations(self):
      se_true = pd.DataFrame(
          {
              "larger_than": [False, False, True],
              "same_color": [False, True, False],
              "n_edges": [2, 3, 2],
          }
      )
      se_true.set_index(["larger_than", "same_color"], inplace=True)
  
      se_test = g.partition_edges(
          relations=["larger_than", "same_color"]
      )
  
      assert_frame_equal(se_test, se_true)
  
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 2 for calling the TestPartitionEdges obj's underlying member method code with example usage: obj.test_relations() without return types