def test_normal_sentiment():
malaya.get_available_sentiment_models()
negative_text = 'kerajaan sebenarnya sangat bencikan rakyatnya, minyak naik dan segalanya'
news_sentiment = malaya.deep_sentiment('normal')
assert news_sentiment.predict(negative_text)['negative'] > 0
def test_bahdanau_sentiment():
malaya.get_available_sentiment_models()
negative_text = 'kerajaan sebenarnya sangat bencikan rakyatnya, minyak naik dan segalanya'
news_sentiment = malaya.deep_sentiment('bahdanau')
assert len(news_sentiment.predict(negative_text)['attention']) > 1
def test_fasttext_sentiment_batch():
malaya.get_available_sentiment_models()
negative_text = 'kerajaan sebenarnya sangat bencikan rakyatnya, minyak naik dan segalanya'
news_sentiment = malaya.deep_sentiment('fast-text')
assert len(news_sentiment.predict_batch([negative_text, negative_text
])) > 0
