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