93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
"""Tests for payload schemas"""
|
|
|
|
import pytest
|
|
from datetime import datetime
|
|
from sentiment_engine.schemas.payload import NormalizedPayload, SourceType, AssetMention, EngagementMetrics
|
|
|
|
|
|
class TestNormalizedPayload:
|
|
"""Tests for NormalizedPayload schema"""
|
|
|
|
def test_valid_payload(self):
|
|
payload = NormalizedPayload(
|
|
source_id="test_source",
|
|
source_type=SourceType.NEWS,
|
|
source_credibility_base=0.8,
|
|
ingest_ts=datetime.now().timestamp(),
|
|
publish_ts=datetime.now().timestamp(),
|
|
asset_mentions=[AssetMention(asset_id="BTC", mention_span=(0, 3), confidence=0.9, source_text="BTC", mention_type="ticker")],
|
|
raw_text="BTC surges to new highs",
|
|
title="BTC Surges",
|
|
url="https://test.com",
|
|
author="Test Author",
|
|
content_length=100,
|
|
language="en"
|
|
)
|
|
assert payload.source_id == "test_source"
|
|
assert payload.has_assets is True
|
|
assert payload.get_assets() == ["BTC"]
|
|
|
|
def test_empty_text_raises(self):
|
|
with pytest.raises(ValueError, match="raw_text cannot be empty"):
|
|
NormalizedPayload(
|
|
source_id="test",
|
|
source_type=SourceType.NEWS,
|
|
source_credibility_base=0.5,
|
|
ingest_ts=1234567890.0,
|
|
raw_text="",
|
|
content_length=0,
|
|
language="en"
|
|
)
|
|
|
|
def test_whitespace_text_raises(self):
|
|
with pytest.raises(ValueError, match="raw_text cannot be empty"):
|
|
NormalizedPayload(
|
|
source_id="test",
|
|
source_type=SourceType.NEWS,
|
|
source_credibility_base=0.5,
|
|
ingest_ts=1234567890.0,
|
|
raw_text=" ",
|
|
content_length=0,
|
|
language="en"
|
|
)
|
|
|
|
|
|
class TestAssetMention:
|
|
"""Tests for AssetMention schema"""
|
|
|
|
def test_valid_mention(self):
|
|
mention = AssetMention(
|
|
asset_id="BTC",
|
|
mention_span=(0, 3),
|
|
confidence=0.9,
|
|
source_text="BTC",
|
|
mention_type="ticker"
|
|
)
|
|
assert mention.asset_id == "BTC"
|
|
assert mention.confidence == 0.9
|
|
|
|
|
|
class TestEngagementMetrics:
|
|
"""Tests for EngagementMetrics"""
|
|
|
|
def test_total_engagement(self):
|
|
metrics = EngagementMetrics(retweets=10, likes=50, replies=5, upvotes=100, comments=20)
|
|
assert metrics.total_engagement() == 185
|
|
|
|
def test_default_zero(self):
|
|
metrics = EngagementMetrics()
|
|
assert metrics.total_engagement() == 0
|
|
|
|
|
|
class TestSourceType:
|
|
"""Tests for SourceType enum"""
|
|
|
|
def test_all_values(self):
|
|
assert SourceType.NEWS == "news"
|
|
assert SourceType.SOCIAL == "social"
|
|
assert SourceType.EXCHANGE_ANN == "exchange_ann"
|
|
assert SourceType.REGULATORY == "regulatory"
|
|
assert SourceType.CORPORATE == "corporate"
|
|
assert SourceType.FORUM == "forum"
|
|
assert SourceType.ON_CHAIN == "on_chain"
|