Skip GSD and size filtering without altitude

This commit is contained in:
Roman Meshko
2026-04-19 21:50:48 +03:00
parent 7d897df380
commit 5cfcdb5fd5
9 changed files with 54 additions and 11 deletions
+9
View File
@@ -5,6 +5,15 @@ def test_ai_config_from_dict_defaults():
assert cfg.model_batch_size == 8
assert cfg.frame_period_recognition == 4
assert cfg.frame_recognition_seconds == 2
assert cfg.has_altitude is False
def test_ai_config_from_dict_altitude_override_sets_flag():
from inference import ai_config_from_dict
cfg = ai_config_from_dict({"altitude": 400})
assert cfg.has_altitude is True
assert cfg.altitude == 400
def test_ai_config_from_dict_overrides():
+17
View File
@@ -118,6 +118,23 @@ def test_resolve_media_for_detect_override_wins():
assert "paths" not in cfg
def test_resolve_media_for_detect_omits_altitude_when_not_provided():
# Arrange
import main
tm = main.TokenManager(_access_jwt(), "")
mock_ann = MagicMock()
mock_ann.fetch_user_ai_settings.return_value = {"probabilityThreshold": 0.2}
mock_ann.fetch_media_path.return_value = "/m/v.mp4"
with patch("main.annotations_client", mock_ann):
# Act
cfg, path = main._resolve_media_for_detect("vid-2", tm, None)
# Assert
assert "altitude" not in cfg
assert cfg["probability_threshold"] == 0.2
assert path == "/m/v.mp4"
def test_resolve_media_for_detect_raises_when_no_media_path():
# Arrange
import main