Subversion Repositories Local Hare Voting

#!/p/python/python

import os

import random

import cStringIO

import parseinput

import tally

import vote

def testfile(dirpath, filename):

    print filename + ":",

    tally.fdOut = cStringIO.StringIO()

    filenameCSV = os.path.join(dirpath, filename)

    filenameOut = os.path.splitext(filenameCSV)[0] + ".out"

    nWinners, ballots = parseinput.parsefile(filenameCSV)

    random.seed(1)

    vote.domany(10, nWinners, ballots)

    tally.trace("")

    vote.doone(nWinners, ballots)

    result = tally.fdOut.getvalue()

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

    for dirpath, dirnames, filenames in os.walk(dir):

        for filename in filenames:

            if filename.endswith(".csv"):

                testfile(dirpath, filename)

if __name__=="__main__":

    main("tests")