def test_calculate_rating_for_player_weights(self):
# test non-equal weights
self.rating1.weight = 1
self.rating1.save()
self.rating2.weight = 2
self.rating2.save()
self.assertEqual(decimal.Decimal('3.0'), rating.get_weight_total_for_set(self.rankset_empty))
rating.PlayerRatingSelection.objects.create(player=self.player1, rankset=self.rankset_empty,
rating=self.rating1, selection=self.rating1.options.get(value=1))
rating.PlayerRatingSelection.objects.create(player=self.player1, rankset=self.rankset_empty,
rating=self.rating2, selection=self.rating2.options.get(value=1))
rating.PlayerRatingSelection.objects.create(player=self.player1, rankset=self.rankset_empty,
rating=self.rating2, selection=self.rating2.options.get(value=2))
rating.PlayerRatingSelection.objects.create(player=self.player1, rankset=self.rankset_empty,
rating=self.rating2, selection=self.rating2.options.get(value=3))
rating.calculate_rating_for_player(self.player1, self.rankset_empty)
player_rating = rating.get_rating_for_player(self.player1, self.rankset_empty)
self.assertEqual(decimal.Decimal('4.3333'), player_rating)
# set the weights back
self.rating1.weight = 1
self.rating1.save()
self.rating2.weight = 1
self.rating2.save()
self.assertEqual(decimal.Decimal('2.0'), rating.get_weight_total_for_set(self.rankset_empty))
def test_get_weight_total_for_set(self):
self.assertEqual(decimal.Decimal('2.0'), rating.get_weight_total_for_set(self.rankset_empty))
