You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

17 lines
481 B
Python

# This entrypoint file to be used in development. Start by reading README.md
import prob_calculator
from unittest import main
prob_calculator.random.seed(95)
hat = prob_calculator.Hat(blue=4, red=2, green=6)
probability = prob_calculator.experiment(
hat=hat,
expected_balls={"blue": 2,
"red": 1},
num_balls_drawn=4,
num_experiments=3000)
print("Probability:", probability)
# Run unit tests automatically
main(module='test_module', exit=False)