blob: dc2d0266c7b2ec3b0635d3ab679bc8f6e3f723ba (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
"""Tests for news-related SQLAlchemy models."""
from shared.sa_models import MarketSentimentRow, NewsItemRow, StockSelectionRow, SymbolScoreRow
def test_news_item_row_tablename():
assert NewsItemRow.__tablename__ == "news_items"
def test_symbol_score_row_tablename():
assert SymbolScoreRow.__tablename__ == "symbol_scores"
def test_market_sentiment_row_tablename():
assert MarketSentimentRow.__tablename__ == "market_sentiment"
def test_stock_selection_row_tablename():
assert StockSelectionRow.__tablename__ == "stock_selections"
def test_news_item_row_columns():
cols = {c.name for c in NewsItemRow.__table__.columns}
assert cols >= {"id", "source", "headline", "published_at", "sentiment", "category"}
def test_symbol_score_row_columns():
cols = {c.name for c in SymbolScoreRow.__table__.columns}
assert cols >= {"id", "symbol", "news_score", "composite", "updated_at"}
|
