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