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"