def test_simple_score(self): crit1 = ScoreCriteria('crit', 50) crit1.populate(10) crit2 = ScoreCriteria('crit2', 10) crit2.objects = ['crit5', 'crit10'] set1 = AffinitySet(crit1,None) set2 = AffinitySet(crit2,None) self.assertEqual(100, set1.getAffinity(set2)) self.assertEqual(20, set2.getAffinity(set1))
def test_blacklist(self): crit = ScoreCriteria('crit', 0) crit.populate(10) crit.blacklist(['crit1', 'crit5', 'crit9']) self.assertEqual(crit.objects, ['crit2', 'crit3', 'crit4', 'crit6', 'crit7', 'crit8', 'crit10'])