Subversion Repositories Remote Hare Voting

#!/p/python/python

import os
import random

import parseinput
import tally

def testfile(dirpath, filename):
    print filename + ":",
    filenameCSV = os.path.join(dirpath, filename)
    filenameOut = os.path.splitext(filenameCSV)[0] + ".out"
    result = tallyfile(filenameCSV)
    handleresult(result, filename, filenameOut)

def tallyfile(filenameCSV):
    nWinners, ballots = parseinput.parsefile(filenameCSV)
    random.seed(1)
    return str(tally.dotally(nWinners, ballots)) + "\n"

def handleresult(result, filename, filenameOut):
    if os.path.isfile(filenameOut):
        fOut = open(filenameOut)
        correct = fOut.read()
        if result == correct:
            print "OK"
        else:
            print "***** MISMATCH *****"
            print "      EXPECTED:", correct
            print "      GOT:     ", result
    else:
        print "***** Creating output for " + filename + ": " + result,
        fOut = open(filenameOut, "w")
        fOut.write(result)

def main(dir):
    tally.fTrace = 0
    for dirpath, dirnames, filenames in os.walk(dir):
        for filename in filenames:
            if filename.endswith(".csv"):
                testfile(dirpath, filename)

if __name__=="__main__":
    main("tests")