def test_negative_removal(self):
eID = None
query = "foo and bar"
rss = RefinableSearchState(eID, query, self.dk_iqr, self.classifier_config, self.mdb_info, "results")
rss.register_negative_feedback(1)
rss.remove_negative_feedback(1)
self.assertSetEqual(rss.negatives, set())
def test_setting_negatives(self):
eID = None
query = "foo and bar"
rss = RefinableSearchState(eID, query, self.dk_iqr, self.classifier_config, self.mdb_info, "results")
assert rss.negatives == set()
rss.register_negative_feedback(1)
assert rss.negatives == set((1,))
def test_negative_removal(self):
eID = None
query = 'foo and bar'
rss = RefinableSearchState(eID, query, self.dk_iqr,
self.classifier_config, self.mdb_info,
'results')
rss.register_negative_feedback(1)
rss.remove_negative_feedback(1)
self.assertSetEqual(rss.negatives, set())
def test_setting_negatives(self):
eID = None
query = 'foo and bar'
rss = RefinableSearchState(eID, query, self.dk_iqr,
self.classifier_config, self.mdb_info,
'results')
assert rss.negatives == set()
rss.register_negative_feedback(1)
assert rss.negatives == set((1, ))
def test_refinement_side_effect(self):
eID = None
query = "foo and bar"
pos_set = set(range(0, 10))
neg_set = set(range(10, 20))
rss = RefinableSearchState(eID, query, self.dk_iqr, self.classifier_config, self.mdb_info, "results")
rss.register_positive_feedback(pos_set)
rss.register_negative_feedback(neg_set)
rss.register_positive_feedback(10)
self.assertNotIn(10, rss.negatives)
self.assertIn(10, rss.positives)
rss.register_negative_feedback(0)
self.assertNotIn(0, rss.positives)
self.assertIn(0, rss.negatives)
def test_refinement_side_effect(self):
eID = None
query = 'foo and bar'
pos_set = set(range(0, 10))
neg_set = set(range(10, 20))
rss = RefinableSearchState(eID, query, self.dk_iqr,
self.classifier_config, self.mdb_info,
'results')
rss.register_positive_feedback(pos_set)
rss.register_negative_feedback(neg_set)
rss.register_positive_feedback(10)
self.assertNotIn(10, rss.negatives)
self.assertIn(10, rss.positives)
rss.register_negative_feedback(0)
self.assertNotIn(0, rss.positives)
self.assertIn(0, rss.negatives)
def test_refinement_inheritance(self):
eID = None
query = "foo and bar"
pos_set = set((23, 25, 27, 60, 123, 227, 457, 967))
neg_set = set((794, 986, 5967, 7254, 23456, 82456))
rss = RefinableSearchState(eID, query, self.dk_iqr, self.classifier_config, self.mdb_info, "results")
rss.register_positive_feedback(pos_set)
rss.register_negative_feedback(neg_set)
rss2 = RefinableSearchState(rss)
self.assertSetEqual(rss.positives, rss2.positives)
self.assertSetEqual(rss2.positives, pos_set)
self.assertSetEqual(rss.negatives, rss2.negatives)
self.assertSetEqual(rss2.negatives, neg_set)
rss.register_positive_feedback(0)
rss2.register_positive_feedback(1)
self.assertNotIn(0, rss2.positives)
self.assertNotIn(1, rss.positives)
rss.register_negative_feedback(2)
rss2.register_negative_feedback(3)
self.assertNotIn(2, rss2.negatives)
self.assertNotIn(3, rss.negatives)
def test_refinement_inheritance(self):
eID = None
query = 'foo and bar'
pos_set = set((23, 25, 27, 60, 123, 227, 457, 967))
neg_set = set((794, 986, 5967, 7254, 23456, 82456))
rss = RefinableSearchState(eID, query, self.dk_iqr,
self.classifier_config, self.mdb_info,
'results')
rss.register_positive_feedback(pos_set)
rss.register_negative_feedback(neg_set)
rss2 = RefinableSearchState(rss)
self.assertSetEqual(rss.positives, rss2.positives)
self.assertSetEqual(rss2.positives, pos_set)
self.assertSetEqual(rss.negatives, rss2.negatives)
self.assertSetEqual(rss2.negatives, neg_set)
rss.register_positive_feedback(0)
rss2.register_positive_feedback(1)
self.assertNotIn(0, rss2.positives)
self.assertNotIn(1, rss.positives)
rss.register_negative_feedback(2)
rss2.register_negative_feedback(3)
self.assertNotIn(2, rss2.negatives)
self.assertNotIn(3, rss.negatives)
