def test_recommend(self):
userID = 'Leopoldo Pires'
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals(['You, Me and Dupree', 'Just My Luck'],
recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals(['You, Me and Dupree', 'Just My Luck'],
recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(
DictDataModel({
'John': {
'A': 5.0,
'B': 3.0,
'C': 2.0
},
'Mark': {
'A': 3.0,
'B': 4.0
},
'Lucy': {
'B': 2.0,
'C': 5.0
}
}), True, False, False)
self.assertEquals(['A'], recSys.recommend('Lucy', 4))
users2 = {
"Amy": {
"Dr. Dog": 4,
"Lady Gaga": 3,
"Phoenix": 4
},
"Ben": {
"Dr. Dog": 5,
"Lady Gaga": 2
},
"Clara": {
"Lady Gaga": 3.5,
"Phoenix": 4
}
}
recSys = SlopeOneRecommender(DictDataModel(users2), True, False, False)
self.assertEquals(['Phoenix'], recSys.recommend('Ben', 1))
def test_recommend(self):
userID = "Leopoldo Pires"
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals(["You, Me and Dupree", "Just My Luck"], recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals(["You, Me and Dupree", "Just My Luck"], recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(
DictDataModel(
{"John": {"A": 5.0, "B": 3.0, "C": 2.0}, "Mark": {"A": 3.0, "B": 4.0}, "Lucy": {"B": 2.0, "C": 5.0}}
),
True,
False,
False,
)
self.assertEquals(["A"], recSys.recommend("Lucy", 4))
users2 = {
"Amy": {"Dr. Dog": 4, "Lady Gaga": 3, "Phoenix": 4},
"Ben": {"Dr. Dog": 5, "Lady Gaga": 2},
"Clara": {"Lady Gaga": 3.5, "Phoenix": 4},
}
recSys = SlopeOneRecommender(DictDataModel(users2), True, False, False)
self.assertEquals(["Phoenix"], recSys.recommend("Ben", 1))
def test_semi_recommend(self):
userID = "Leopoldo Pires"
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals(["You, Me and Dupree"], recSys.recommend(userID, 1))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals(["You, Me and Dupree"], recSys.recommend(userID, 1))
def test_full_recommend(self):
userID = "Maria Gabriela"
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals([], recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals([], recSys.recommend(userID, 4))
def test_empty_recommend(self):
userID = "Marcel Caraciolo"
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals([], recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals([], recSys.recommend(userID, 4))
def test_semi_recommend(self):
userID = 'Leopoldo Pires'
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals(['You, Me and Dupree'], recSys.recommend(userID, 1))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals(['You, Me and Dupree'], recSys.recommend(userID, 1))
def test_full_recommend(self):
userID = 'Maria Gabriela'
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals([], recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals([], recSys.recommend(userID, 4))
def test_empty_recommend(self):
userID = 'Marcel Caraciolo'
recSys = SlopeOneRecommender(self.model, True, False, False)
self.assertEquals([], recSys.recommend(userID, 4))
recSys = SlopeOneRecommender(self.model, True, False, True)
self.assertEquals([], recSys.recommend(userID, 4))
