99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
"""
|
|
tests/test_wheeling.py
|
|
-----------------------
|
|
Tests for core/wheeling.py — full-cover wheel generation.
|
|
"""
|
|
|
|
import pytest
|
|
from core.wheeling import wheel_full, wheel_count, MAX_TICKETS
|
|
|
|
|
|
# ── wheel_count ───────────────────────────────────────────────────────────────
|
|
|
|
def test_wheel_count_basic():
|
|
assert wheel_count([1, 2, 3, 4, 5, 6], 5) == 6 # C(6,5)
|
|
|
|
def test_wheel_count_exact_pick():
|
|
assert wheel_count([1, 2, 3, 4, 5], 5) == 1 # C(5,5)
|
|
|
|
def test_wheel_count_larger():
|
|
assert wheel_count(list(range(1, 10)), 5) == 126 # C(9,5)
|
|
|
|
def test_wheel_count_k_zero_returns_zero():
|
|
assert wheel_count([1, 2, 3, 4, 5], 0) == 0
|
|
|
|
def test_wheel_count_k_exceeds_n_returns_zero():
|
|
assert wheel_count([1, 2, 3], 5) == 0
|
|
|
|
def test_wheel_count_deduplicates_input():
|
|
assert wheel_count([1, 1, 2, 3, 4, 5], 5) == 1 # C(5,5) after dedup
|
|
|
|
|
|
# ── wheel_full — valid cases ──────────────────────────────────────────────────
|
|
|
|
def test_wheel_full_ticket_count():
|
|
tickets = wheel_full([1, 2, 3, 4, 5, 6], 5)
|
|
assert len(tickets) == 6
|
|
|
|
def test_wheel_full_single_ticket():
|
|
tickets = wheel_full([5, 14, 22, 36, 69], 5)
|
|
assert len(tickets) == 1
|
|
assert tickets[0] == [5, 14, 22, 36, 69]
|
|
|
|
def test_wheel_full_each_ticket_sorted():
|
|
tickets = wheel_full([10, 3, 7, 1, 5, 2], 4)
|
|
for t in tickets:
|
|
assert t == sorted(t)
|
|
|
|
def test_wheel_full_no_duplicate_tickets():
|
|
tickets = wheel_full(list(range(1, 9)), 5) # C(8,5) = 56
|
|
as_tuples = [tuple(t) for t in tickets]
|
|
assert len(as_tuples) == len(set(as_tuples))
|
|
|
|
def test_wheel_full_all_numbers_in_pool():
|
|
pool = [5, 14, 22, 36, 55, 69]
|
|
tickets = wheel_full(pool, 4)
|
|
for t in tickets:
|
|
for n in t:
|
|
assert n in pool
|
|
|
|
def test_wheel_full_deduplicates_input():
|
|
# [1,1,2,3,4,5] → pool [1,2,3,4,5] → C(5,5) = 1
|
|
tickets = wheel_full([1, 1, 2, 3, 4, 5], 5)
|
|
assert len(tickets) == 1
|
|
|
|
def test_wheel_full_at_cap(monkeypatch):
|
|
import core.wheeling as wm
|
|
monkeypatch.setattr(wm, "MAX_TICKETS", 56)
|
|
# C(8,5) = 56 exactly at cap — should pass
|
|
tickets = wm.wheel_full(list(range(1, 9)), 5)
|
|
assert len(tickets) == 56
|
|
|
|
def test_wheel_full_returns_list_of_lists():
|
|
tickets = wheel_full([1, 2, 3, 4, 5, 6], 5)
|
|
assert isinstance(tickets, list)
|
|
for t in tickets:
|
|
assert isinstance(t, list)
|
|
|
|
|
|
# ── wheel_full — error cases ──────────────────────────────────────────────────
|
|
|
|
def test_wheel_full_k_zero_raises():
|
|
with pytest.raises(ValueError, match="at least 1"):
|
|
wheel_full([1, 2, 3, 4, 5], 0)
|
|
|
|
def test_wheel_full_k_exceeds_n_raises():
|
|
with pytest.raises(ValueError, match="exceeds"):
|
|
wheel_full([1, 2, 3], 5)
|
|
|
|
def test_wheel_full_exceeds_cap_raises():
|
|
# C(10,5) = 252 > MAX_TICKETS (200)
|
|
with pytest.raises(ValueError, match="252"):
|
|
wheel_full(list(range(1, 11)), 5)
|
|
|
|
def test_wheel_full_error_message_includes_count():
|
|
with pytest.raises(ValueError) as exc_info:
|
|
wheel_full(list(range(1, 11)), 5)
|
|
assert "252" in str(exc_info.value)
|
|
assert str(MAX_TICKETS) in str(exc_info.value)
|