def get_most_read(key):
"""
Gets the most read news from a given page (national or international)
:param key: the key of the source page (e.g: g1, localDF, WP, etc.)
:return: a list with the most read news from the page
"""
# Check if the News Source is national or internacional
if key in news_source_national.strategies:
return news_source_national.get_most_read(key)
elif key in news_source_international.strategies:
return news_source_international.get_most_read(key)
elif key in news_source_sports.strategies:
return news_source_sports.get_most_read(key)
elif key in news_source_entertainment.strategies:
return news_source_entertainment.get_most_read(key)
elif key in news_source_technology.strategies:
return news_source_technology.get_most_read(key)
elif key in news_source_music.strategies:
return news_source_music.get_most_read(key)
elif key in news_source_reddit.strategies:
return news_source_reddit.get_most_read(key)
else:
raise ValueError('No news source found for SOURCE={0}'.format(key))
def test_whiplash(self):
news, title = nsm.get_most_read('m_whiplash')
self.assertEqual(len(news), 10)
def test_rs(self):
news, title = nsm.get_most_read('m_rs')
self.assertEqual(len(news), 5)
