def test_update_vote(): value = 3 for voter_id, _, _ in EXISTING_VOTES: Vote.update(voter_id, voted_id=value) votes = Vote.find_all() for vote in votes: assert vote.voted_id == value
def count_vote(room_id): votes = Vote.find_all(room_id=room_id) tally = Counter() for vote in votes: tally[vote.voted_id] += 1 return tally, len(votes)
def test_find_all_votes(): votes = Vote.find_all(room_id=1) assert len(votes) == 2 for vote in votes: print(vote.voter_id, vote.room_id, vote.voted_id)