def test_same_article(): arts = [ Article(url='dummy.com', body="Canada and USA.", title="Canada and USA.") ] states = [ "yes" ] algo = KeywordAlgorithm() algo.train(arts, states) assert math.ceil(algo.score(arts)[0]) == math.ceil(ScoreMapper().map_score("yes"))
def test_existing_keywords(): arts = [ Article(url='dummy.com', body="Canada.", title="Canada."), Article(url='dummy.com', body="USA.", title="USA."), Article(url='dummy.com', body="Mexico.", title="Mexico.") ] states = [ "yes", "read", "no" ] algo = KeywordAlgorithm() algo.train(arts, states) test_art = Article(url='dummy.com', body="USA.", title="USA.") assert math.floor(algo.score([test_art])[0]) == math.ceil(ScoreMapper().map_score("read"))