Exemplo n.º 1
0
    def test_set_recommendation_impact_via_redis(self):

        user_info = {'uid': '1', 'remote_ip': '127.0.0.1', 'uri': 'testuri'}

        records_to_test = range(0, 100000)
        recommendations = {90000: 1.0, 4300: 0.3}

        self.redis.set("obelix::recommendations::1",
                       json.dumps(recommendations))

        settings = ObelixSearchEngineSettings(redis=self.redis)
        settings_json = settings.dump()

        settings_json['recommendations_impact'] = 1
        self.redis.set("obelix::settings", json.dumps(settings_json))

        records, scores = rank_records_obelix(user_info,
                                              records_to_test,
                                              rg=10,
                                              jrec=0,
                                              settings=settings)

        for key, val in recommendations.iteritems():
            self.assertTrue(key in records)

        settings_json['recommendations_impact'] = 0.1
        self.redis.set("obelix::settings", json.dumps(settings_json))

        records, scores = rank_records_obelix(user_info,
                                              records_to_test,
                                              rg=10,
                                              jrec=0,
                                              settings=settings)
        for key, val in recommendations.iteritems():
            self.assertFalse(key in records)
Exemplo n.º 2
0
    def test_set_recommendation_impact_via_redis(self):

        user_info = {'uid': '1', 'remote_ip': '127.0.0.1', 'uri': 'testuri'}

        records_to_test = range(0, 100000)
        recommendations = {90000: 1.0, 4300: 0.3}

        self.redis.set("obelix::recommendations::1", json.dumps(recommendations))

        settings = ObelixSearchEngineSettings(redis=self.redis)
        settings_json = settings.dump()

        settings_json['recommendations_impact'] = 1
        self.redis.set("obelix::settings", json.dumps(settings_json))

        records, scores = rank_records_obelix(user_info,
                                               records_to_test,
                                               rg=10, jrec=0,
                                               settings=settings)

        for key, val in recommendations.iteritems():
            self.assertTrue(key in records)

        settings_json['recommendations_impact'] = 0.1
        self.redis.set("obelix::settings", json.dumps(settings_json))

        records, scores = rank_records_obelix(user_info,
                                               records_to_test,
                                               rg=10, jrec=0,
                                               settings=settings)
        for key, val in recommendations.iteritems():
            self.assertFalse(key in records)