def test_article_less_than_operator():
article_1 = Article(date.fromisoformat('2020-03-15'), None, None, None,
None)
article_2 = Article(date.fromisoformat('2020-04-20'), None, None, None,
None)
assert article_1 < article_2
def load_articles_and_tags(data_path: str, repo: MemoryRepository):
tags = dict()
for data_row in read_csv_file(os.path.join(data_path,
'news_articles.csv')):
article_key = int(data_row[0])
number_of_tags = len(data_row) - 6
article_tags = data_row[-number_of_tags:]
# Add any new tags; associate the current article with tags.
for tag in article_tags:
if tag not in tags.keys():
tags[tag] = list()
tags[tag].append(article_key)
del data_row[-number_of_tags:]
# Create Article object.
article = Article(date=date.fromisoformat(data_row[1]),
title=data_row[2],
first_para=data_row[3],
hyperlink=data_row[4],
image_hyperlink=data_row[5],
id=article_key)
# Add the Article to the repository.
repo.add_article(article)
# Create Tag objects, associate them with Articles and add them to the repository.
for tag_name in tags.keys():
tag = Tag(tag_name)
for article_id in tags[tag_name]:
article = repo.get_article(article_id)
make_tag_association(article, tag)
repo.add_tag(tag)
def test_tag_construction(tag):
assert tag.tag_name == 'New Zealand'
for article in tag.tagged_articles:
assert False
assert not tag.is_applied_to(Article(None, None, None, None, None, None))
def article():
return Article(
date.fromisoformat('2020-03-15'),
'Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room',
'The self-isolation deadline has been pushed back',
'https://www.nzherald.co.nz/business/news/article.cfm?c_id=3&objectid=12316800',
'https://th.bing.com/th/id/OIP.0lCxLKfDnOyswQCF9rcv7AHaCz?w=344&h=132&c=7&o=5&pid=1.7'
)
def make_article():
article = Article(
article_date, "Coronavirus: First case of virus in New Zealand",
"The first case of coronavirus has been confirmed in New Zealand and authorities are now scrambling to track down people who may have come into contact with the patient.",
"https://www.stuff.co.nz/national/health/119899280/ministry-of-health-gives-latest-update-on-novel-coronavirus",
"https://resources.stuff.co.nz/content/dam/images/1/z/e/3/w/n/image.related.StuffLandscapeSixteenByNine.1240x700.1zduvk.png/1583369866749.jpg"
)
return article
def make_article(new_article_date):
article = Article(
new_article_date,
'Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room',
'The self-isolation deadline has been pushed back',
'https://www.nzherald.co.nz/business/news/article.cfm?c_id=3&objectid=12316800',
'https://th.bing.com/th/id/OIP.0lCxLKfDnOyswQCF9rcv7AHaCz?w=344&h=132&c=7&o=5&pid=1.7'
)
return article
def test_make_tag_associations():
article = Article(
date.fromisoformat('2020-03-15'),
'Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room',
'The self-isolation deadline has been pushed back',
'https://www.nzherald.co.nz/business/news/article.cfm?c_id=3&objectid=12316800',
'https://th.bing.com/th/id/OIP.0lCxLKfDnOyswQCF9rcv7AHaCz?w=344&h=132&c=7&o=5&pid=1.7'
)
tag = Tag('New Zealand')
make_tag_association(article, tag)
# Check that the Article knows about the Tag.
assert article.is_tagged()
assert article.is_tagged_by(tag)
# check that the Tag knows about the Article.
assert tag.is_applied_to(article)
assert article in tag.tagged_articles
def test_repository_can_add_article(in_memory_repo):
article = Article(
date.fromisoformat('2020-03-09'),
'Second US coronavirus cruise tests negative amid delays and cancellations',
'It was revealed ...',
'https://www.nzherald.co.nz/travel/news/article.cfm?c_id=7&objectid=12315024',
'https://www.nzherald.co.nz/resizer/ix7hy3lzkMWUkD8hE6kdZ-8oaOM=/620x349/smart/filters:quality(70)/arc-anglerfish-syd-prod-nzme.s3.amazonaws.com/public/7VFOBLCBCNDHLICBY3CTPFR2L4.jpg',
7)
in_memory_repo.add_article(article)
assert in_memory_repo.get_article(7) is article
def test_repository_can_add_article(session_factory):
repo = SqlAlchemyRepository(session_factory)
number_of_articles = repo.get_number_of_articles()
new_article_id = number_of_articles + 1
article = Article(
date.fromisoformat('2020-03-09'),
'Second US coronavirus cruise tests negative amid delays and cancellations',
'It was revealed ...',
'https://www.nzherald.co.nz/travel/news/article.cfm?c_id=7&objectid=12315024',
'https://www.nzherald.co.nz/resizer/ix7hy3lzkMWUkD8hE6kdZ-8oaOM=/620x349/smart/filters:quality(70)/arc-anglerfish-syd-prod-nzme.s3.amazonaws.com/public/7VFOBLCBCNDHLICBY3CTPFR2L4.jpg',
new_article_id
)
repo.add_article(article)
assert repo.get_article(new_article_id) == article
def get_articles_by_date(self, target_date: date) -> List[Article]:
target_article = Article(date=target_date,
title=None,
first_para=None,
hyperlink=None,
image_hyperlink=None)
matching_articles = list()
try:
index = self.article_index(target_article)
for article in self._articles[index:None]:
if article.date == target_date:
matching_articles.append(article)
else:
break
except ValueError:
# No articles for specified date. Simply return an empty list.
pass
return matching_articles
def test_article_construction():
article = Article(
date.fromisoformat('2020-03-15'),
'Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room',
'The self-isolation deadline has been pushed back',
'https://www.nzherald.co.nz/business/news/article.cfm?c_id=3&objectid=12316800',
'https://th.bing.com/th/id/OIP.0lCxLKfDnOyswQCF9rcv7AHaCz?w=344&h=132&c=7&o=5&pid=1.7'
)
assert article.id is None
assert article.date == date.fromisoformat('2020-03-15')
assert article.title == 'Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room'
assert article.first_para == 'The self-isolation deadline has been pushed back'
assert article.hyperlink == 'https://www.nzherald.co.nz/business/news/article.cfm?c_id=3&objectid=12316800'
assert article.image_hyperlink == 'https://th.bing.com/th/id/OIP.0lCxLKfDnOyswQCF9rcv7AHaCz?w=344&h=132&c=7&o=5&pid=1.7'
assert len(article.comments) == 0
assert len(article.tags) == 0
assert repr(
article
) == '<Article 2020-03-15 Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room>'
def load_articles_and_tags(data_path: str, repo: MemoryRepository):
tags = dict()
for data_row in read_csv_file(os.path.join(data_path,
'Data1000Movies.csv')):
article_key = int(data_row[0])
article_tags = data_row[2].split(",")
# Add any new tags; associate the current article with tags.
for tag in article_tags:
if tag not in tags.keys():
tags[tag] = list()
tags[tag].append(article_key)
# Create Article object.
article = Article(
date=int(data_row[6]),
title=data_row[1],
first_para=data_row[3],
director=data_row[4],
actors=data_row[5],
#hyperlink=data_row[4],
#image_hyperlink=data_row[5],
id=article_key)
# Add the Article to the repository.
repo.add_article(article)
# Create Tag objects, associate them with Articles and add them to the repository.
for tag_name in tags.keys():
tag = Tag(tag_name)
for article_id in tags[tag_name]:
article = repo.get_article(article_id)
make_tag_association(article, tag)
repo.add_tag(tag)
def test_make_comment_establishes_relationships():
comment_text = 'COVID-19 in the USA!'
user = User('bspringsteen', '0987654321')
article = Article(
date.fromisoformat('2020-03-15'),
'Coronavirus travel restrictions: Self-isolation deadline pushed back to give airlines breathing room',
'The self-isolation deadline has been pushed back',
'https://www.nzherald.co.nz/business/news/article.cfm?c_id=3&objectid=12316800',
'https://th.bing.com/th/id/OIP.0lCxLKfDnOyswQCF9rcv7AHaCz?w=344&h=132&c=7&o=5&pid=1.7'
)
comment = make_comment(comment_text, user, article)
# Check that the User object knows about the Comment.
assert comment in user.comments
# Check that the Comment knows about the User.
assert comment.user is user
# Check that Article knows about the Comment.
assert comment in article.comments
# Check that the Comment knows about the Article.
assert comment.article is article
def test_tag_construction():
tag = Tag('New Zealand')
assert tag.tag_name == 'New Zealand'
assert len(tag.tagged_articles) == 0
assert not tag.is_applied_to(Article(None, None, None, None, None, None))
def dict_to_article(dict):
article = Article(dict.id, dict.date, dict.title, dict.first_para,
dict.hyperlink)
# Note there's no comments or tags.
return article
