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"))
