diff --git a/dice_stats.py b/dice_stats.py index 83a99cb..91fdf8e 100644 --- a/dice_stats.py +++ b/dice_stats.py @@ -19,6 +19,20 @@ class FiveDice: return False return True + def is_three_of_a_kind(self): + for face in self.faces(): + if self.faces.count(face) >= 3: + return True + return False + + def is_four_of_a_kind(self): + for face in self.faces(): + if self.faces.count(face) >= 4: + return True + return False + + + dice = FiveDice() successes = 0 trials = 1000000 diff --git a/die.py b/die.py index 14580bb..00ee769 100644 --- a/die.py +++ b/die.py @@ -15,3 +15,4 @@ class Die: def roll(self): self.face = randint(1, 6) return self.face +