def test_redis_set_and_get_with_prefix_encoder(self):
storage = RedisStorage(RedisMock(), prefix='pre::', encoder=json)
storage.set("theKey", "theValue")
storage.set("theKey2", "theValue2")
assert storage.get("theKey2") == "theValue2"
assert storage.get("theKey") == "theValue"
assert storage.get("noKey") == None
def test_redis_set_and_get_with_prefix_encoder(self):
storage = RedisStorage(RedisMock(), prefix='pre::', encoder=json)
storage.set("theKey", "theValue")
storage.set("theKey2", "theValue2")
assert storage.get("theKey2") == "theValue2"
assert storage.get("theKey") == "theValue"
assert storage.get("noKey") == None
class TestObelix(unittest.TestCase):
def setUp(self):
""" setup any state tied to the execution of the given method in a
class. setup_method is invoked for every test method of a class.
"""
self.cache = RedisStorage(RedisMock(), prefix='pre::', encoder=json)
self.recommendations = RedisStorage(RedisMock(), 'recommendations::')
self.queues = RedisQueue(RedisMock(), encoder=json)
self.obelix = Obelix(self.cache, self.recommendations, self.queues)
def test_obelix_settings_init(self):
obelix = Obelix(self.cache, self.recommendations, self.queues)
conf = obelix.config
assert conf['recommendations_impact'] == 0.5
assert conf['score_lower_limit'] == 0.2
assert conf['score_upper_limit'] == 10
assert conf['score_min_limit'] == 10
assert conf['score_min_multiply'] == 4
assert conf['score_one_result'] == 1
assert conf['method_switch_limit'] == 20
def test_obelix_settings_parameter(self):
# Set settings
settings = {'recommendations_impact': 5.5,
'score_lower_limit': 2.2,
'score_upper_limit': 1.5,
'score_min_limit': 9,
'score_min_multiply': 7,
'score_one_result': 3,
'method_switch_limit': 13}
# storage.set('settings', settings)
obelix = Obelix(self.cache, self.recommendations,
self.queues, settings)
conf = obelix.config
assert isinstance(conf['recommendations_impact'], float)
for key, value in settings.items():
assert conf[key] == value
def test_rank_records_no_recommendations(self):
obelix = self.obelix
uid = 1
hitset = range(1, 30)
# Set recommendations
# None
# Page one
result1 = obelix.rank_records(hitset, uid, 10, 0)
# Page two
result2 = obelix.rank_records(hitset, uid, 10, 11)
# Page 3
result3 = obelix.rank_records(hitset, uid, 10, 21)
# TODO Check result
pre1 = ([29, 28, 27, 26, 25, 24, 23, 22, 21, 20],
[0.8, 0.7724137931034483, 0.7448275862068965,
0.7172413793103448, 0.6896551724137931, 0.6620689655172414,
0.6344827586206896, 0.6068965517241379, 0.5793103448275861,
0.5517241379310345])
pre2 = ([19, 18, 17, 16, 15, 14, 13, 12, 11, 10],
[0.5241379310344827, 0.4965517241379309, 0.46896551724137925,
0.4413793103448276, 0.4137931034482758, 0.386206896551724,
0.35862068965517235, 0.3310344827586207,
0.3034482758620689, 0.2758620689655171])
pre3 = ([9, 8, 7, 6, 5, 4, 3, 2, 1],
[0.24827586206896535, 0.2206896551724138, 0.193103448275862,
0.16551724137931023, 0.13793103448275845,
0.1103448275862069, 0.08275862068965512,
0.055172413793103336, 0.027586206896551557])
assert pre1 == result1
assert pre2 == result2
assert pre3 == result3
def test_rank_records_with_two_recommendations(self):
obelix = self.obelix
uid = 1
hitset = range(1, 30)
# Set recommendations
pre_reco = {5: 0.5, 20: 1.0}
self.recommendations.set(uid, pre_reco)
# Page one
result1 = obelix.rank_records(hitset, uid, 10, 0)
# Page two
result2 = obelix.rank_records(hitset, uid, 10, 11)
# Page 3
result3 = obelix.rank_records(hitset, uid, 10, 21)
# TODO Check results by hand
pre1 = ([20, 29, 28, 27, 26, 25, 24, 5, 23, 22],
[0.7758620689655172, 0.4, 0.38620689655172413,
0.37241379310344824, 0.3586206896551724,
0.3448275862068966, 0.3310344827586207, 0.3189655172413792,
0.3172413793103448, 0.30344827586206896])
pre2 = ([21, 19, 18, 17, 16, 15, 14, 13, 12, 11],
[0.28965517241379307, 0.26206896551724135, 0.24827586206896546,
0.23448275862068962, 0.2206896551724138,
0.2068965517241379, 0.193103448275862, 0.17931034482758618,
0.16551724137931034, 0.15172413793103445])
pre3 = ([10, 9, 8, 7, 6, 4, 3, 2, 1],
[0.13793103448275856, 0.12413793103448267, 0.1103448275862069,
0.096551724137931, 0.08275862068965512,
0.05517241379310345, 0.04137931034482756,
0.027586206896551668, 0.013793103448275779])
assert pre1 == result1
assert pre2 == result2
assert pre3 == result3
def test_rank_records_with_one_record(self):
obelix = self.obelix
uid = 1
hitset = [8]
# Set recommendations
pre_reco = {5: 0.5, 20: 1.0}
self.recommendations.set(uid, pre_reco)
obelix.config['score_one_result'] = 1
# Page one
result1 = obelix.rank_records(hitset, uid, 10, 0)
# Page two
result2 = obelix.rank_records(hitset, uid, 10, 11)
# Page 3
result3 = obelix.rank_records(hitset, uid, 10, 21)
# TODO Check results by hand
pre1 = ([8], [0.5])
pre2 = ([], [])
pre3 = ([], [])
assert pre1 == result1
assert pre2 == result2
assert pre3 == result3
def test_rank_records_with_no_record(self):
obelix = self.obelix
uid = 1
hitset = []
# Set recommendations
pre_reco = {5: 0.5, 20: 1.0}
self.recommendations.set(uid, pre_reco)
obelix.config['score_one_result'] = 1
# Page one
result1 = obelix.rank_records(hitset, uid, 10, 0)
# Page two
result2 = obelix.rank_records(hitset, uid, 10, 11)
# Page 3
result3 = obelix.rank_records(hitset, uid, 10, 21)
# TODO Check results by hand
pre1 = ([], [])
pre2 = ([], [])
pre3 = ([], [])
assert pre1 == result1
assert pre2 == result2
assert pre3 == result3