16 lines
449 B
Python
16 lines
449 B
Python
|
# This entrypoint file to be used in development. Start by reading README.md
|
||
|
|
import prob_calculator
|
||
|
|
from unittest import main
|
||
|
|
|
||
|
|
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)
|