diff options
Diffstat (limited to 'tests/test_data.py')
| -rw-r--r-- | tests/test_data.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/test_data.py b/tests/test_data.py index 2fb131e..a5f91ce 100644 --- a/tests/test_data.py +++ b/tests/test_data.py @@ -75,11 +75,13 @@ def test_make_batch_shapes_and_padding(): def test_flagged_composites_are_composite(): - # sanity: the rank-4 sieve signature set is composite, has no divisors <= 7, - # and covers the probe's candidate window [102, 211] (Addendum 3/6) + # sanity: the rank-4 sieve signature set is composite, has no divisors <= 7 from src.eval import _sieve_rank_signature sig = _sieve_rank_signature(4, 101, 200) - assert sig == {121, 143, 169, 187, 209}, f"unexpected rank-4 signature: {sig}" + assert sig == {121, 143, 169, 187}, f"unexpected rank-4 signature: {sig}" for n in sig: assert any(n % d == 0 for d in range(2, int(n ** 0.5) + 1)) assert all(n % d != 0 for d in (2, 3, 5, 7)) + # 209 = 11*19 is also rank-4 but outside [101,200]; verify it appears in wider range + sig_wide = _sieve_rank_signature(4, 101, 210) + assert 209 in sig_wide |