diff --git a/__pycache__/die.cpython-310.pyc b/__pycache__/die.cpython-310.pyc new file mode 100644 index 0000000..bbfa34d Binary files /dev/null and b/__pycache__/die.cpython-310.pyc differ diff --git a/dice_stats.py b/dice_stats.py index 83a99cb..aee761d 100644 --- a/dice_stats.py +++ b/dice_stats.py @@ -1,5 +1,6 @@ from die import Die from tqdm import tqdm +from collections import Counter class FiveDice: def __init__(self): @@ -18,6 +19,21 @@ class FiveDice: if face != 1: return False return True + + + def is_three_of_a_kind(self): + count = Counter(self.faces()) + for face_count in count.values(): + if face_count == 3: + return True + return False + + def is_four_of_a_kind(self): + count = Counter(self.faces()) + for face_count in count.values(): + if face_count == 4: + return True + return False dice = FiveDice() successes = 0 @@ -28,6 +44,9 @@ for trial in tqdm(range(trials)): successes += 1 print(successes/trials) - +#print(f"dice rolls: {dice.faces()}") +print(dice.faces()) +print(dice.is_three_of_a_kind()) +print(dice.is_four_of_a_kind())