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)
