r2b2.athena

Athena audit module.

Module Contents

class r2b2.athena.Athena(alpha: float, delta: float, max_fraction_to_draw: float, contest: r2b2.contest.Contest)[source]

Bases: r2b2.audit.Audit

Athena audit implementation.

An Athena audit is a type of risk-limiting audit that accounts for round-by-round auditor decisions. For a given sample size (in the context of a round schedule), the audit software calculates a minimum number of votes for the reported winner that must be found in the sample to stop the audit and confirm the reported outcome.

Variables
  • alpha (float) – Risk limit. Alpha represents the chance that, given an incorrectly called election, the audit will fail to force a full recount.

  • max_fraction_to_draw (float) – The maximum number of ballots the auditors are willing to draw as a fraction of the ballots in the contest.

  • delta (float) – Delta value.

Initialize an Athena audit.

get_min_sample_size(self, sub_audit: r2b2.audit.PairwiseAudit, min_sprob: float = 10 ** - 6)[source]

Computes the minimum sample size that has a stopping size (kmin). Here we find a practical minimum instead of the theoretical minimum (BRAVO’s minimum) to avoid floating-point imprecisions in the later convolution process.

Parameters
  • sub_audit (PairwiseAudit) – Compute minimum sample size for this subaudit.

  • min_sprob (float) – Round sizes with below min_sprob stopping probability are excluded.

Returns

int – The minimum sample size of the audit, adherent to the min_sprob.

satisfactory_sample_size(self, left, right, sprob, num_dist, denom_dist)[source]

Helper method that returns True if the round size satisfies the stopping probability.

next_sample_size(self, *args, **kwargs)[source]

Generate estimates of possible next sample sizes.

Note: To be used during live/interactive audit execution.

stopping_condition_pairwise(self, pair: str, verbose: bool = False)bool[source]

Check, without finding the kmin, whether the audit is complete.

Parameters

pair (str) – Dictionary key referencing pairwise subaudit. Evaluate the stopping condition for this subaudit.

Returns

bool – Whether or not the pairwise stopping condition has been met.

next_min_winner_ballots_pairwise(self, sub_audit: r2b2.audit.PairwiseAudit)int[source]

Compute stopping size for a given subaudit.

Parameters

sub_audit (PairwiseAudit) – Compute next stopping size for this subaudit.

Returns

int – Stopping size for most recent round.

compute_min_winner_ballots(self, sub_audit: r2b2.audit.PairwiseAudit, rounds: List[int], *args, **kwargs)[source]

Compute the minimum number of winner ballots for a round schedule.

Extend the audit’s round schedule with the passed (partial) round schedule, and then extend the audit’s minimum number of winner ballots schedule with the corresponding minimums to meet the stopping condition.

Parameters
  • sub_audit (PairwiseAudit) – Compute minimum winner ballots for this Pairwise subaudit.

  • rounds (List[int]) – A (partial) round schedule of the audit.

find_kmin(self, sub_audit: r2b2.audit.PairwiseAudit, sample_size: int, append: bool)[source]

Search for a kmin (minimum number of winner ballots) satisfying all stopping criteria.

Parameters
  • sub_audit (PairwiseAudit) – Find kmin for this subaudit.

  • sample_size (int) – Sample size to find kmin for.

  • append (bool) – Optionally append the kmins to the min_winner_ballots list. This may not always be desirable here because, for example, appending happens automatically outside this method during an interactive audit.

compute_all_min_winner_ballots(self, sub_audit: r2b2.audit.PairwiseAudit, max_sample_size: int = None, *args, **kwargs)[source]

Compute the minimum number of winner ballots for the complete (that is, ballot-by-ballot) round schedule.

Note: Due to limited convolutional precision, results may be off somewhat after the

stopping probability very nearly equals 1.

Parameters
  • sub_audit (PairwiseAudit) – Compute minimum winner ballots for this pairwise subaudit.

  • max_sample_size (int) – Optionally set the maximum sample size to generate stopping sizes (kmins) up to. If not provided the maximum sample size is determined by max_frac_to_draw and the total contest ballots.

Returns

None, kmins are appended to the min_winner_ballots list.

compute_risk(self, votes_for_winner: int, pair: str, *args, **kwargs)[source]

Return the hypothetical (Minerva) p-value if votes_for_winner were obtained in the most recent round.

get_risk_level(self)[source]

Return the risk level of an interactive Athena audit.

Non-interactive and bulk Athena audits are not considered here since the sampled number of reported winner ballots is not available.

__repr__(self)

String representation of Audit class.

Note

Can (and perhaps should) be overwritten in subclass.

__str__(self)

Human readable string (i.e. printable) representation of Audit class.

Note

Can (and perhaps should) be overwritten in subclass.

current_dist_null(self)

Update distribution_null for each sub audit for current round.

_current_dist_null_pairwise(self, sub_audit: PairwiseAudit, bulk_use_round_size=False)

Update distribution_null for a single PairwiseAudit

Parameters
  • sub_audit (PairwiseAudit) – Pairwise subaudit for which to update distribution.

  • bulk_use_round_size (bool) – Optional argument used by bulk methods. Since the bulk methods do not sample ballots for candidates during the rounds, this flag simply uses the round schedule as the round draw (instead of the pairwise round draw) for updating the distribution. Default is False.

current_dist_reported(self)

Update distribution_reported_tally for each subaudit for current round.

_current_dist_reported_pairwise(self, sub_audit: PairwiseAudit, bulk_use_round_size=False)

Update dist_reported for a single PairwiseAudit.

Parameters
  • sub_audit (PairwiseAudit) – Pairwise subaudit for which to update distriution.

  • bulk_use_round_size (bool) – Optional argument used by bulk methods. Since the bulk methods do not sample ballots for candidates during the rounds, this flag simply uses the round schedule as the round draw (instead of the pairwise round draw) for updating the distribution. Default is False.

truncate_dist_null(self)

Update risk schedule and truncate null distribution for each subaudit.

_truncate_dist_null_pairwise(self, pair: str)

Update risk schedule and truncate null distribution for a single subaudit.

Parameters

pair (str) – Dictionary key for subaudit (within the audit’s subaudits) to truncate distribution and update risk schedule.

truncate_dist_reported(self)

Update stopping prob schedule and truncate reported distribution for each subaudit.

_truncate_dist_reported_pairwise(self, pair)

Update stopping prob schedule and truncate reported distribution for a single subaudit.

Parameters

pair (str) – Dictionary key for subaudit (within the audit’s subaudits) to truncate distribution and update stopping prob schedule.

__get_interval(self, dist: List[float])

Get relevant interval [l, u] of given distribution.

Find levels l and u such that cdf(l) < tolerance and 1 - cdf(u) < tolerance. The purpose of this is to improve efficiency in the current_dist_* functions for audits without replacement where almost all of the hypergeometric distribution falls in a fraction of its range, i.e. between l and u.

Note

cdf() in this context does not require cdf(infinity) = 1, although the distribution should sum very closely to 1.

asn(self, pair: str)

Compute ASN as described in BRAVO paper for pair of candidates.

Given the fractional margin for the reported winner and the risk limit (alpha) produce the average number of ballots sampled during the audit.

Parameters

pair (str) – Dictionary key referencing pairwise audit in audit’s subaudits.

Returns

int – ASN value.

execute_round(self, sample_size: int, sample: dict, verbose: bool = False)bool

Execute a single, non-interactive audit round.

Executes 1 round of the audit, given its current state. If the audit is stopped, its state will not be modified.

Warning: This method assumes the audit is in the correct state to be executed. If multiple

executions of a full audit will be run the same audit object, make sure to call reset on the audit object between full executions.

Parameters
  • sample_size (int) – Total ballots sampled by the end of this round (cumulative).

  • sample (dict) – Sample counts for each candidate by the end of this round (cumulative).

Returns

bool – True if the audit met its stopping condition by this round.

run(self, verbose: bool = False)

Begin interactive audit execution.

Begins the interactive version of the audit. While computations for different audits will vary, the process for executing each one is the same. This provides a process for selecting a sample size, determining if the ballots found for the reported winner in that sample size meet the stopping condition(s), and if not continuing with the audit. As the audit proceeds, data including round sizes, ballots for the winner in each round size, and per round risk and stopping probability are stored.

_reset(self)

Reset attributes modified during run().

stopping_condition(self, verbose: bool = False)bool

Determine if the audits stopping condition has been met.

The audit stopping condition is met if and only if each pairwise stopping condition is met.

next_min_winner_ballots(self, verbose: bool = False)

Compute next stopping size of given (actual) sample sizes for all subaudits.