def test_flood_find_areas(self): areas = areas_finder.get_areas(Image.open(self._data("black.png")), 100) self.assertEqual(len(areas), 1) areas = areas_finder.get_areas(Image.open(self._data("test.png")), 100) self.assertGreater(len(areas), 3) areas = areas_finder.get_areas(Image.open(self._data("test03.png")), 100) self.assertGreater(len(areas), 2)
def test_get_triangles_for_area(self): s = self.get_seeder() source_im = Image.open(os.path.join(data_folder, "black.png")) areas = areas_finder.get_areas(source_im, 1000) triangles = s.get_triangles_for_area(areas[0]) self.assertEqual(len(triangles), 2) self.assertEqual(triangles[0].coordinates, [0, 0, 0, 64, 64, 64]) self.assertEqual(triangles[1].coordinates, [0, 0, 64, 0, 64, 64])