diff --git a/tests/lottery_predictor/test_draw.py b/tests/lottery_predictor/test_draw.py new file mode 100644 index 0000000..1a7ae6d --- /dev/null +++ b/tests/lottery_predictor/test_draw.py @@ -0,0 +1,10 @@ +import pytest + +import lottery_predictor.draw as draw + + +def test_generate_random_ticket() -> None: + assert isinstance(draw.generate_random_ticket(max_main=69, max_game=26), tuple) + assert isinstance(draw.generate_random_ticket(max_main=70, max_game=24), tuple) + with pytest.raises(TypeError): + draw.generate_random_ticket()