class DetectionIle(TestCase):
def setUp(self):
self.analyseImageWorld = AnalyseImageWorld()
self.analyseImageWorld.chargerImage("images/table2/detection1.png")
self.detection = DetectionIles(self.analyseImageWorld.image, 2)
def test_detecter(self):
self.detection.detecter()
nombreTresor = len(self.detection.ilesIdentifiees)
self.assertEqual(nombreTresor, 16)
def test_trouverContourIles(self):
c, h = self.detection.trouverContoursIles("Rouge")
bool = len(c) >= 4
self.assertTrue(bool)
def test_trouverIles(self):
c, h = self.detection.trouverContoursIles("Rouge")
self.detection.trouverIles(c, h, "Rouge")
bool = len(self.detection.ilesIdentifiees) == len(c)
self.assertTrue(bool)
def test_eleminerCoutoursNegligeable(self):
c, h = self.detection.trouverContoursIles("Rouge")
c2 = self.detection.eleminerCoutoursNegligeable(c, h)
bool = len(c2) <= len(c)
self.assertTrue(bool)
def trouverIles(self):
print("\nDetection des iles...")
detectionMultipleIles = []
for i in range(NB_DETECTION_TOT):
print 'detection: %d sur 20' % i
self.chargerImage()
detectionIles = DetectionIles(self.image, self.stationBase.getNumTable())
detectionIles.detecter()
listIles = self.eliminerContoursProcheRobot(detectionIles.getIlesIdentifiees())
detectionMultipleIles.append(listIles)
time.sleep(0.01)
listIles = self.resultatPlusCommun(detectionMultipleIles)
self.stationBase.carte.setIles(listIles)
def setUp(self):
self.analyseImageWorld = AnalyseImageWorld()
self.analyseImageWorld.chargerImage("images/table2/detection1.png")
self.detection = DetectionIles(self.analyseImageWorld.image, 2)
def setUp(self):
self.image = self.imageCamera = cv2.imread(
ConfigPath.Config().appendToProjectPath('images/table2/detection1.png'))
self.detectionElements = DetectionIles(self.image)
class TestDetectionElementsCartographiques(TestCase):
def setUp(self):
self.image = self.imageCamera = cv2.imread(
ConfigPath.Config().appendToProjectPath('images/table2/detection1.png'))
self.detectionElements = DetectionIles(self.image)
def test_detecte_bon_nombre_de_formes_jaune(self):
self.detectionElements.detecter()
nombreJaune = self.detectionElements._getNombreIleCouleur("Jaune")
self.assertEqual(nombreJaune, 4)
def test_detecte_bon_nombre_de_formes_bleue(self):
self.detectionElements.detecter()
nombreBleu = self.detectionElements._getNombreIleCouleur("Bleu")
self.assertEqual(nombreBleu, 4)
def test_detecte_bon_nombre_de_formes_verte(self):
self.detectionElements.detecter()
nombreVert = self.detectionElements._getNombreIleCouleur("Vert")
self.assertEqual(nombreVert, 4)
def test_detecte_bon_nombre_de_formes_rouge(self):
self.detectionElements.detecter()
nombreRouge = self.detectionElements._getNombreIleCouleur("Rouge")
self.assertEqual(nombreRouge, 4)
def test_detecte_bon_nombre_tresor(self):
self.detectionElements.detecterTresor()
tresors = self.detectionElements.tresorIdentifies
self.assertEqual(len(tresors), 3)
