diff --git a/dice_stats.py b/dice_stats.py index 83a99cb..7e64086 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,22 @@ class FiveDice: if face != 1: return False return True + + def is_three_of_a_kind(self): + counts = Counter(self.dice) + for count in counts.value(): + if count >= 3: + return True + return False + + def is_four_of_a_kind(self): + counts = Counter(self.dice) + for count in counts.value(): + if count >= 4: + return True + return False + + dice = FiveDice() successes = 0