Python TwoPlayerTrueSkillCalculator Examples

Example #1

File: UnitTests.py Project: lianhaoqu/PSkills

	def setUp(self):
		self.calculator = TwoPlayerTrueSkillCalculator()
		self.gameInfo = defaultGameInfo()

Example #2

File: UnitTests.py Project: lianhaoqu/PSkills

class TwoPlayerTrueSkillCalculatorTests(unittest.TestCase):
	def setUp(self):
		self.calculator = TwoPlayerTrueSkillCalculator()
		self.gameInfo = defaultGameInfo()
		
	def assertRating(self, expectedMean, expectedStandardDeviation, actual):
		self.assertAlmostEqual(actual.mean, expectedMean, delta = _errorTolerance)
		self.assertAlmostEqual(actual.standardDeviation, expectedStandardDeviation, delta = _errorTolerance)
		
	def assertMatchQuality(self, expectedMatchQuality, actualMatchQuality):
		self.assertAlmostEqual(expectedMatchQuality, actualMatchQuality, delta = 0.0005)
	
	def test_twoPlayerTestDrawn(self):
		player1 = Player(1)
		player2 = Player(2)
		gameInfo = defaultGameInfo()
	
		team1 = Team(player1, gameInfo.defaultRating)
		team2 = Team(player2, gameInfo.defaultRating)
		teams = [team1, team2]
	
		newRatings = self.calculator.calculateNewRatings(gameInfo, teams, [1, 1])
	
		for newRating in newRatings:
			self.assertRating(25.0, 6.4575196623173081, newRating[1])
		self.assertMatchQuality(0.447, self.calculator.calculateMatchQuality(gameInfo, teams))
	
	def test_oneOnOneMassiveUpsetDrawTest(self):
		player1 = Player(1)
		player2 = Player(2)
	
		team1 = Team()
		team1.addPlayer(player1, self.gameInfo.defaultRating)
		team2 = Team()
		team2.addPlayer(player2, Rating(50, 12.5))
		teams = [team1, team2]
	
		newRatingsWinLose = self.calculator.calculateNewRatings(self.gameInfo, teams, [1, 1])
	
		for newRating in newRatingsWinLose:
			player = newRating[0]
			if player == player1:
				self.assertRating(31.662, 7.137, newRating[1])
			else:
				self.assertRating(35.010, 7.910, newRating[1])
		self.assertMatchQuality(.110, self.calculator.calculateMatchQuality(self.gameInfo, teams))
	
	def test_twoPlayerTestNotDrawn(self):
		player1 = Player(1)
		player2 = Player(2)
	
		team1 = Team(player1, self.gameInfo.defaultRating)
		team2 = Team(player2, self.gameInfo.defaultRating)
		teams = [team1, team2]
	
		newRatings = self.calculator.calculateNewRatings(self.gameInfo, teams, [1, 2])
	
		for player, rating in newRatings:
			if player == player1:
				self.assertRating(29.39583201999924, 7.171475587326186, rating)
			else:
				self.assertRating(20.60416798000076, 7.171475587326186, rating)
		self.assertMatchQuality(0.447, self.calculator.calculateMatchQuality(self.gameInfo, teams))
	
	def test_twoPlayerChessTestNotDrawn(self):
		player1 = Player(1)
		player2 = Player(2)
		gameInfo = GameInfo(1200.0, 1200.0 / 3.0, 200.0, 1200.0 / 300.0, 0.03)
	
		team1 = Team(player1, Rating(1301.0007, 42.9232))
		team2 = Team(player2, Rating(1188.7560, 42.5570))
	
		newRatings = self.calculator.calculateNewRatings(gameInfo, [team1, team2], [1, 2])
	
		for newRating in newRatings:
			player = newRating[0]
			if player == player1:
				self.assertRating(1304.7820836053318, 42.843513887848658, newRating[1])
			else:
				self.assertRating(1185.0383099003536, 42.485604606897752, newRating[1])