system-prompts-and-models-o.../dealix/tests/unit/test_sentiment.py
2026-05-01 14:03:52 +03:00

34 lines
822 B
Python

"""Unit tests for Arabic sentiment."""
from __future__ import annotations
from dealix.intelligence.sentiment import ArabicSentiment
def test_positive():
s = ArabicSentiment()
r = s.analyze("الخدمة ممتازة وكفو")
assert r.label == "positive"
assert r.score > 0
def test_negative():
s = ArabicSentiment()
r = s.analyze("التجربة كانت سيئة وفاشلة")
assert r.label == "negative"
assert r.score < 0
def test_neutral():
s = ArabicSentiment()
r = s.analyze("السلام عليكم")
assert r.label == "neutral"
def test_negator_flips_sign():
s = ArabicSentiment()
r_pos = s.analyze("الخدمة ممتازة")
r_neg = s.analyze("الخدمة ما ممتازة")
assert r_pos.score > 0
assert r_neg.score < r_pos.score