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'])