2020-09-29 19:13:24 +02:00
|
|
|
# This entrypoint file to be used in development. Start by reading README.md
|
|
|
|
|
import prob_calculator
|
|
|
|
|
from unittest import main
|
|
|
|
|
|
2021-07-16 17:32:09 +02:00
|
|
|
prob_calculator.random.seed(95)
|
2020-09-29 19:13:24 +02:00
|
|
|
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)
|
