def checkImage(self, directory, filename): print("Testing image {} ".format(filename)) name = os.path.splitext(filename)[0] expectedStr = name.split("__")[1] expectedList = [int(x) for x in expectedStr] fullFilename = os.path.join(directory, filename) scanner.setCameraDebugValues() actualResult = scanner.getVotesFromImage(fullFilename) if(actualResult != expectedList): cv2.waitKey(10001) self.assertEqual(actualResult, expectedList, msg="\nFailure for test image: {}".format(fullFilename)) print("Success for" + filename)
def checkImage(self, directory, filename): print("Testing image {} ".format(filename)) name = os.path.splitext(filename)[0] expectedStr = name.split("__")[1] expectedList = [int(x) for x in expectedStr] fullFilename = os.path.join(directory, filename) scanner.setCameraDebugValues() actualResult = scanner.getVotesFromImage(fullFilename) if (actualResult != expectedList): cv2.waitKey(10001) self.assertEqual( actualResult, expectedList, msg="\nFailure for test image: {}".format(fullFilename)) print("Success for" + filename)
print('___TAKING PICTURE___') #img = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) result = getVotesFromCV2Img(frame) voteCount += 1 processResult(result, totalResult, voteCount, f) if k == 27: #27 on mac 1048603: #ESC cam.release() break f.close() def processResult(result, totalResult, voteCount, f): print(result) for c, r in enumerate(result): totalResult[c+1] = totalResult[c+1] + r f.write(' '.join(str(x) for x in result) + '\n') sorted_x = sorted(totalResult.items(), key=operator.itemgetter(1)) print("Vote Count: {} ".format(voteCount)) for c, r in sorted_x: print("{:>3} : {:>4} ".format(c,r)) if __name__ == '__main__': print(len(sys.argv)) if len(sys.argv) >= 2: print(sys.argv[1]) if sys.argv[1] == 'debug': setCameraDebugValues() main()