code: def test_partition(self, st_a, st_b, expected):
      result = st_a.partition == st_b.partition
      self.assertEqual(result, expected)
  
imports: from geomstats.test.parametrizers import DataBasedParametrizer
  from geomstats.test.test_case import TestCase
  from geomstats.test_cases.geometry.stratified.point_set import PointSetTestCase, PointTestCase
  from .data.wald_space import SplitTestData, TopologyTestData, WaldSpaceTestData, WaldTestData
label_desc: Use image node_id 1 for calling the TestTopology obj's underlying member method code with example usage: obj.test_partition(st_a, st_b, expected) without return types