def rt_helper(prefix, num_booths, expected):
    params_filename = prefix + ".json"
    params = simulate.setup_params(params_filename, num_booths)
    expected = [eval(row[0]) for row in csv.reader(open(prefix + "-exact-expected-" + str(num_booths) + ".txt"))]

    error_str = "Results of trial {} do not match.  Expected: {}     Got: {}"
    for t in range(params["num_trials"]):
        actual = simulate.simulate_election_day(params)
        print((t, expected[t], actual))
        if expected[t] != actual:
            pytest.fail(error_str.format(t, expected[t], actual))
def sed_helper(params_filename, num_booths, expected):
    params = simulate.setup_params(params_filename, num_booths)
    actual = simulate.simulate_election_day(params)
    if (actual != expected):
        pytest.fail("Expected:{}  Got: {}". format(actual, expected))
def sed_helper(params_filename, num_booths, expected):
    params = simulate.setup_params(params_filename, num_booths)
    actual = simulate.simulate_election_day(params)
    assert actual == expected
Example #4
0

import simulate
#import priority_queue

#params = simulate.setup_params('data/params2.json', 8)
#Expects True

#params = simulate.setup_params('data/params2.json', 6)
#Expects False

#params = simulate.setup_params('data/params0.json', 1)
#Expects False

params = simulate.setup_params('data/params0.json', 2)
#Expects False

s = simulate.simulate_election_day(params)
print(s)
def rt_helper(params_filename, num_booths, expected):
    params = simulate.setup_params(params_filename, num_booths)
    actual = simulate.run_trials(params)
    if abs(actual - expected) > 0.000001:
        pytest.fail("Actual and expected results do not match.  {:.5f} != {:.5f}".format(actual, expected))