code: def test_feature(self):
      sv_true = pd.DataFrame(
          {"color": ["b", "g", "r"], "n_nodes": [1, 3, 1]}
      )
      sv_true.set_index("color", inplace=True)
  
      se_true = pd.DataFrame(
          {
              "color_s": ["b", "g", "g", "g"],
              "color_t": ["g", "b", "g", "r"],
              "n_edges": [1, 2, 3, 1],
          }
      )
      se_true.set_index(["color_s", "color_t"], inplace=True)
  
      sv_test, se_test = g.partition_graph("color")
  
      assert_frame_equal(sv_test, sv_true)
      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 1 for calling the TestPartitionGraph obj's underlying member method code with example usage: obj.test_feature() without return types