code: def partition_test_data(self):
      # TODO: add false example
      smoke_data = [
          dict(
              st_a=Topology(
                  n_labels=3,
                  partition=((1, 0), (2,)),
                  split_sets=((), ()),
              ),
              st_b=Topology(
                  n_labels=3,
                  partition=((2,), (0, 1)),
                  split_sets=((), ()),
              ),
              expected=True,
          ),
          dict(
              st_a=Topology(
                  n_labels=3,
                  partition=((1,), (0,), (2,)),
                  split_sets=((), (), ()),
              ),
              st_b=Topology(
                  n_labels=3,
                  partition=((0,), (1,), (2,)),
                  split_sets=((), (), ()),
              ),
              expected=True,
          ),
      ]
  
      return self.generate_tests(smoke_data)
  
imports: import random
  import geomstats.backend
  from geomstats.geometry.stratified.wald_space import Split, Topology, Wald, WaldSpace
  from geomstats.test.data import TestData
  from geomstats.test.test_case import np_backend
  from .point_set import PointSetTestData, PointTestData
label_desc: Use image node_id 1 for calling the TopologyTestData obj's underlying member method code with example usage: obj.partition_test_data() and returns: self