def test_multiply_weight_float(self): """ Tests multiplying node weight. """ fraction = 0.5 bd = Board(30, 30) coords = [[15, 15], [2, 7], [9, 25]] weights = [20, 2, 100] for coord, weight in zip(coords, weights): bd.setWeight(coord, weight) bd.multiplyWeight(coord, fraction) self.assertEqual(bd.getWeight(coord), weight * fraction)
def test_multiply_weight_int(self): """ Tests multiplying node weight. """ multiplier = 5 bd = Board(30, 30) coords = [[15, 15], [2, 7], [9, 25]] weights = [6, 14, 90] for coord, weight in zip(coords, weights): bd.setWeight(coord, weight) bd.multiplyWeight(coord, multiplier) for coord, weight in zip(coords, weights): desiredWeight = weight * multiplier if desiredWeight < -100: desiredWeight = -100 elif desiredWeight > 100: desiredWeight = 100 self.assertEqual(bd.getWeight(coord), desiredWeight)