def test_same_article(): arts = [ Article(url='dummy.com', body="Canada and USA.", title="Canada and USA.") ] states = [ "yes" ] algo = CosineAlgorithm() algo.train(arts, states) assert math.ceil(algo.score_single_article(arts[0])) == math.ceil(ScoreMapper().map_score("yes"))
def test_different_article(): arts = [ Article(url='dummy.com', body="Canada and USA.", title="Canada and USA.") ] states = [ "yes" ] algo = CosineAlgorithm() algo.train(arts, states) test_art = Article(url='dummy.com', body="Trinidad and Tobago.", title="Trinidad and Tobago.") assert math.floor(algo.score_single_article(test_art)) == 0