37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""场景生成：源数量、频道、位置与接收半径满足题面范围。"""
|
|
|
|
import math
|
|
|
|
from referee import DEFAULT_SPEC, mixed_scene, random_omni_scene
|
|
|
|
|
|
def test_random_omni_scene_within_spec():
|
|
for seed in range(5):
|
|
sources = random_omni_scene(seed=seed)
|
|
assert 10 <= len(sources) <= 16 # 源总数范围 [题面]
|
|
channels = [s.channel for s in sources]
|
|
assert len(set(channels)) == len(channels) # 每频道至多一个源
|
|
assert all(1 <= c <= 20 for c in channels)
|
|
for s in sources:
|
|
assert math.hypot(s.x, s.y) <= DEFAULT_SPEC.field.omega_radius_m
|
|
assert 1000 <= s.r_eff_m <= 1500
|
|
assert not s.directional
|
|
|
|
|
|
def test_scene_is_reproducible_by_seed():
|
|
a = random_omni_scene(seed=123, count=12)
|
|
b = random_omni_scene(seed=123, count=12)
|
|
assert a == b
|
|
|
|
|
|
def test_mixed_scene_directional_count():
|
|
sources = mixed_scene(seed=7, count=12, directional=4)
|
|
assert sum(1 for s in sources if s.directional) == 4
|
|
assert all(0 <= s.heading_deg < 360 for s in sources if s.directional)
|
|
|
|
|
|
def test_count_is_validated():
|
|
import pytest
|
|
with pytest.raises(ValueError):
|
|
random_omni_scene(seed=1, count=0)
|