FilesExpand file tree

 master

/

StateFunction.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (42 loc) · 2.13 KB

 master

/

StateFunction.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (42 loc) · 2.13 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

import random

from Candidate import Candidate

from PollingData import PollingData

class State:

def __init__(self, name: str, polling_data=None, population=None):

self.name = name

self.polling_data = polling_data or PollingData()

self.population = population or 250

def get_winner_random(self, candidates: [Candidate]) -> Candidate:

"""Chooses a random candidate as winner with uniform distribution"""

return random.choice(candidates)

def get_vote(self, candidates: [Candidate], distribution: [float]) -> Candidate:

rand_float = random.random() # Random percentage

running_total = 0

for i in range(len(distribution)):

running_total += distribution[i]

if rand_float < running_total:

return candidates[i]

return Candidate('Write-in', 'I') # All the other votes must be write-ins

def get_winner_with_distribution(self, candidates: [Candidate], distribution: [float]) -> Candidate:

candidate_sums = {}

for _ in range(self.population):

vote = self.get_vote(candidates, distribution)

if vote in candidate_sums:

candidate_sums[vote] += 1

else:

candidate_sums[vote] = 1

winner = max(candidate_sums, key=candidate_sums.get)

if winner.party not in ['D', 'R']:

print(f'\n\t{winner} won {self.name}, with distribution:')

print('\t', candidate_sums)

print('\tPercentage:',

{key: f'{round(100 * value / sum(candidate_sums.values()), 2)}%' for key, value in candidate_sums.items()})

print('\tPolling Data:', {can: val for can, val in zip(candidates, distribution)})

return winner

def get_winner(self, candidates: [Candidate]) -> Candidate:

distribution = self.get_polling_distribution(candidates)

return self.get_winner_with_distribution(candidates, distribution)

def get_polling_distribution(self, candidates: [Candidate]) -> [float]:

return self.polling_data.get_polling_distribtion(self.name, candidates)

def __repr__(self):

return self.name