def trouverTresors(self):
print("\nDetection des tresors...")
detectionMultipleTresors = []
for i in range(NB_DETECTION_TOT):
print 'detection: %d sur 20' % i
self.chargerImage()
detectionTresors = DetectionTresors(self.image, self.stationBase.getNumTable())
detectionTresors.detecter()
detectionMultipleTresors.append(detectionTresors.getTresorsIdentifies())
time.sleep(0.01)
listTresors = self.resultatPlusCommun(detectionMultipleTresors)
self.stationBase.carte.setTresors(listTresors)
self.debuterDetectionTresors = False
class DetectionTresor(TestCase):
def setUp(self):
self.analyseImageWorld = AnalyseImageWorld()
self.analyseImageWorld.chargerImage('images/table2/detection1.png')
self.detection = DetectionTresors(self.analyseImageWorld.image, 2)
def test_detecter(self):
self.detection.detecter()
nombreTresor = len(self.detection.tresorIdentifies)
self.assertEqual(nombreTresor, 3)
def test_trouverContourTresors(self):
c = self.detection.trouverContoursTresors()
bool = len(c) >= 3
self.assertTrue(bool)
def test_trouverTresors(self):
c = self.detection.trouverContoursTresors()
self.detection.trouverTresors(c)
bool = len(self.detection.tresorIdentifies) == len(c)
self.assertTrue(bool)
def test_eliminerTresorsImpossible(self):
self.detection.tresorIdentifies.append(Tresor(500, 500))
self.detection.eliminerTresorsImpossibles()
bool = self.detection.tresorIdentifies == []
self.assertTrue(bool)
def test_eleminerCoutoursNegligeable(self):
c, h = self.detection.trouverContoursTresors()
c2 = self.detection.eleminerCoutoursNegligeables(c)
bool = len(c2) <= len(c)
self.assertTrue(bool)
def setUp(self):
self.analyseImageWorld = AnalyseImageWorld()
self.analyseImageWorld.chargerImage('images/table2/detection1.png')
self.detection = DetectionTresors(self.analyseImageWorld.image, 2)
