CI Bot
105 lines
3.6 KiB
Python
Executable File
105 lines
3.6 KiB
Python
Executable File
"""classification 模块单元测试."""
|
|
|
|
import pytest
|
|
from domain.classification import (
|
|
AssetClassification,
|
|
AssetLibraryKind,
|
|
ClassificationJob,
|
|
ClassificationJobStatus,
|
|
IngestJobStatus,
|
|
)
|
|
|
|
|
|
class TestAssetLibraryKind:
|
|
"""AssetLibraryKind 枚举测试."""
|
|
|
|
def test_values(self):
|
|
assert AssetLibraryKind.VIDEO == "video"
|
|
assert AssetLibraryKind.VOICE == "voice"
|
|
|
|
|
|
class TestIngestJobStatus:
|
|
"""IngestJobStatus 枚举测试."""
|
|
|
|
def test_values(self):
|
|
assert IngestJobStatus.PENDING == "pending"
|
|
assert IngestJobStatus.PROCESSING == "processing"
|
|
assert IngestJobStatus.COMPLETED == "completed"
|
|
assert IngestJobStatus.FAILED == "failed"
|
|
|
|
|
|
class TestClassificationJobStatus:
|
|
"""ClassificationJobStatus 枚举测试."""
|
|
|
|
def test_values(self):
|
|
assert ClassificationJobStatus.PENDING == "pending"
|
|
assert ClassificationJobStatus.PROCESSING == "processing"
|
|
assert ClassificationJobStatus.COMPLETED == "completed"
|
|
assert ClassificationJobStatus.FAILED == "failed"
|
|
|
|
|
|
class TestAssetClassification:
|
|
"""AssetClassification 枚举测试."""
|
|
|
|
def test_values(self):
|
|
assert AssetClassification.SCENIC == "scenic"
|
|
assert AssetClassification.PRODUCT == "product"
|
|
assert AssetClassification.PERSON == "person"
|
|
assert AssetClassification.ANIMAL == "animal"
|
|
assert AssetClassification.FOOD == "food"
|
|
assert AssetClassification.TECH == "tech"
|
|
assert AssetClassification.SPORT == "sport"
|
|
assert AssetClassification.MUSIC == "music"
|
|
assert AssetClassification.OTHER == "other"
|
|
|
|
|
|
class TestClassificationJobCreate:
|
|
"""ClassificationJob.create 工厂方法测试."""
|
|
|
|
def test_create_with_valid_params(self):
|
|
job = ClassificationJob.create(project_id="proj_001", asset_id="asset_001")
|
|
assert job.id
|
|
assert len(job.id) == 32
|
|
assert job.project_id == "proj_001"
|
|
assert job.asset_id == "asset_001"
|
|
assert job.status == ClassificationJobStatus.PENDING
|
|
assert job.classification == ""
|
|
assert job.confidence == 0.0
|
|
assert job.error_message == ""
|
|
assert job.created_at is not None
|
|
assert job.updated_at is not None
|
|
|
|
def test_create_strips_strings(self):
|
|
job = ClassificationJob.create(
|
|
project_id=" proj_002 ",
|
|
asset_id=" asset_002 ",
|
|
)
|
|
assert job.project_id == "proj_002"
|
|
assert job.asset_id == "asset_002"
|
|
|
|
def test_create_empty_project_id_raises(self):
|
|
with pytest.raises(ValueError, match="project_id"):
|
|
ClassificationJob.create(project_id="", asset_id="a")
|
|
|
|
def test_create_whitespace_project_id_raises(self):
|
|
with pytest.raises(ValueError, match="project_id"):
|
|
ClassificationJob.create(project_id=" ", asset_id="a")
|
|
|
|
def test_create_empty_asset_id_raises(self):
|
|
with pytest.raises(ValueError, match="asset_id"):
|
|
ClassificationJob.create(project_id="p", asset_id="")
|
|
|
|
def test_create_whitespace_asset_id_raises(self):
|
|
with pytest.raises(ValueError, match="asset_id"):
|
|
ClassificationJob.create(project_id="p", asset_id=" ")
|
|
|
|
def test_create_ids_are_unique(self):
|
|
j1 = ClassificationJob.create(project_id="p", asset_id="a")
|
|
j2 = ClassificationJob.create(project_id="p", asset_id="b")
|
|
assert j1.id != j2.id
|
|
|
|
def test_create_timestamps_are_utc(self):
|
|
job = ClassificationJob.create(project_id="p", asset_id="a")
|
|
assert job.created_at.tzinfo is not None
|
|
assert job.updated_at.tzinfo is not None
|