def test_FriendFitnessBad(self): neighbours = [{'id':x} for x in [4,5,6]] fit = fitness.friendFitness(self.cur,neighbours) self.assertEqual(0,fit)
def test_friendFitnessGood(self): neighbours = [{'id':x} for x in [1,2,3]] fit = fitness.friendFitness(self.cur,neighbours) self.assertEqual(1,fit)