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()
