def test_reviews_by_reviewer_maps_reviews_with_same_reviewer():
review1 = review_pb2.Review()
review1.user_id = 101
review2 = review_pb2.Review()
review2.user_id = 101
mapped = reviews_by_reviewer([review1, review2])
assert list(mapped.keys()) == [101]
assert reviews_equal(mapped[101], [review1, review2])
def test_max_date_occurences_chooses_highest_date_count():
review1 = review_pb2.Review()
review1.date = "2000-01-01"
review2 = review_pb2.Review()
review2.date = "2000-02-01"
review3 = review_pb2.Review()
review3.date = "2000-02-01"
assert max_date_occurrences([review1, review2, review3]) == 2
def test_reviews_by_reviewer_maps_reviews_with_different_reviewers():
review1 = review_pb2.Review()
review1.user_id = 101
review2 = review_pb2.Review()
review2.user_id = 102
mapped = reviews_by_reviewer([review1, review2])
assert list(mapped.keys()) == [101, 102]
assert reviews_equal(mapped[101], [review1])
assert reviews_equal(mapped[102], [review2])
def test_formatting_yelp_chi_gives_correct_productid():
test_productid = "rpP9iZsT3NC-Z4pUtQGoiA"
def get_id(map, key):
if key == test_productid:
return 321321
map_service = IDMapService(get_id)
review = review_pb2.Review()
metadata = _get_chi_metadata(productid=test_productid)
_format_yelp_chi_review(review,
metadata=metadata,
productid_map_service=map_service)
assert review.product_id == 321321
def test_formatting_yelp_chi_gives_correct_userid():
test_userid = "IErE0ydkkLfAoePgqrVdUQ"
def get_id(map, key):
if key == test_userid:
return 123123
map_service = IDMapService(get_id)
review = review_pb2.Review()
metadata = _get_chi_metadata(userid=test_userid)
_format_yelp_chi_review(review,
metadata=metadata,
userid_map_service=map_service)
assert review.user_id == 123123
def test_formatting_amazonBook_title_gives_correct_productId():
test_bookTitle = "Crime and Punishment"
def get_id(map, key):
if key == test_bookTitle:
return 42069
map_service = IDMapService(get_id)
review = review_pb2.Review()
test_reviewObj = _get_amazon_reviewobject(title=test_bookTitle)
_format_amazonBook_review(review,
reviewObject=test_reviewObj,
productid_map_service=map_service)
assert review.product_id == 42069
def test_formatting_amazonBook_reviewAuthor_gives_correct_userId():
test_author = "NiallWalsh"
def get_id(map, key):
if key == test_author:
return 1221
map_service = IDMapService(get_id)
review = review_pb2.Review()
test_reviewObj = _get_amazon_reviewobject(author=test_author)
_format_amazonBook_review(review,
reviewObject=test_reviewObj,
userid_map_service=map_service)
assert review.user_id == 1221
def test_formatting_amazonBook_gives_correct_genuine_label():
review = review_pb2.Review()
_format_amazonBook_review(
review, reviewObject=_get_amazon_reviewobject(realclass="2"))
assert review.label == False
def test_max_date_occurrences_counts_review_date_twice():
review1 = review_pb2.Review()
review1.date = "2000-01-01"
review2 = review_pb2.Review()
review2.date = "2000-01-01"
assert max_date_occurrences([review1, review2]) == 2
def test_max_date_occurrences_counts_single_review_date():
review = review_pb2.Review()
review.date = "2000-01-01"
assert max_date_occurrences([review]) == 1
def test_formatting_yelp_chi_gives_correct_review_content():
review = review_pb2.Review()
_format_yelp_chi_review(review, content="I visited")
assert review.review_content == "I visited"
def test_formatting_yelp_nyc_gives_correct_genuine_label():
review = review_pb2.Review()
format_yelp_nyc_review(review, _get_entry_text(),
_get_nyc_metadata(label=1))
assert review.label == False
def test_formatting_yelp_nyc_gives_correct_review_content():
review = review_pb2.Review()
entry = _get_entry_text(content="Blarg")
format_yelp_nyc_review(review, entry, _get_nyc_metadata())
assert review.review_content == "Blarg"
def test_formatting_yelp_nyc_gives_correct_rating():
review = review_pb2.Review()
entry = _get_entry_text()
format_yelp_nyc_review(review, entry, _get_nyc_metadata(rating=1.0))
assert review.rating == 1
def get_review(content="", date="2000-01-01", rating=3.0):
review = review_pb2.Review()
review.review_content = content
review.date = date
review.rating = rating
return review
def test_formatting_yelp_nyc_gives_correct_user_id():
review = review_pb2.Review()
entry = _get_entry_text(userid=2)
format_yelp_nyc_review(review, entry, _get_nyc_metadata())
assert review.user_id == 2
def test_formatting_amazonBook_gives_correct_date():
review = review_pb2.Review()
_format_amazonBook_review(
review, reviewObject=_get_amazon_reviewobject(date="October 19, 2018"))
assert review.date == "10/19/2018"
def test_formatting_amazonBook_gives_correct_review_content():
review = review_pb2.Review()
_format_amazonBook_review(
review,
reviewObject=_get_amazon_reviewobject(review_body="Terrible book"))
assert review.review_content == "Terrible book"
def test_formatting_yelp_chi_gives_correct_genuine_label():
review = review_pb2.Review()
_format_yelp_chi_review(review, metadata=_get_chi_metadata(label="N"))
assert review.label == False
def test_reviews_by_reviewer_maps_single_review():
review = review_pb2.Review()
review.user_id = 101
mapped = reviews_by_reviewer([review])
assert list(mapped.keys()) == [101]
assert reviews_equal(mapped[101], [review])
def test_formatting_yelp_nyc_gives_correct_product_id():
review = review_pb2.Review()
entry = _get_entry_text(productid=3)
format_yelp_nyc_review(review, entry, _get_nyc_metadata())
assert review.product_id == 3
def test_formatting_yelp_nyc_gives_correct_date():
review = review_pb2.Review()
entry = _get_entry_text(date="2001-02-03")
format_yelp_nyc_review(review, entry, _get_nyc_metadata())
assert review.date == "2001-02-03"
def test_formatting_yelp_chi_gives_correct_date():
review = review_pb2.Review()
_format_yelp_chi_review(review,
metadata=_get_chi_metadata(date="2012-12-21"))
assert review.date == "2012-12-21"
