diff --git a/dice_stats.py b/dice_stats.py index 83a99cb..4c0d05e 100644 --- a/dice_stats.py +++ b/dice_stats.py @@ -19,6 +19,36 @@ class FiveDice: return False return True + def is_three_of_a_kind(self): + for face in self.faces(): + if face != face != face: + return False + return True + + def is_four_of_a_kind(self): + for face in self.faces(): + if face != face != face != face: + return False + return True + +dice = FiveDice() +successes = 0 +trials = 1000000 +for trial in tqdm(range(trials)): + dice.roll() + if dice.is_three_of_a_kind(): + successes += 1 +print(successes/trials) + +dice = FiveDice() +successes = 0 +trials = 1000000 +for trial in tqdm(range(trials)): + dice.roll() + if dice.is_four_of_a_kind(): + successes += 1 +print(successes/trials) + dice = FiveDice() successes = 0 trials = 1000000 @@ -26,7 +56,6 @@ for trial in tqdm(range(trials)): dice.roll() if dice.all_ones(): successes += 1 - print(successes/trials)