def assertComponents(self, image, n): topology = Topology() topology.calculate(image) ncomponents = 0 for component in topology.get_components(): ncomponents += 1 self.assertEqual(ncomponents, n)
def test_get_components_height(self): image = self.chess_board() image[self.SIZE/2, :] = 255 topology = Topology() topology.calculate(image) ncomponents = 0 for component in topology.get_components(): ncomponents += 1 self.assertEqual(ncomponents, 2)
def test_single_component(self): image = np.zeros((self.SIZE, self.SIZE)) topology = Topology() topology.calculate(image) self.assertEqual(len(topology), 1)
def assertTopologyLen(self, image, size): topology = Topology() topology.calculate(image) self.assertEqual(len(topology), size)