Gas optimization on average function of Math.sol (#2757)

* change implementation to save gas

* add average test with two max uni256 number
pull/2763/head
rotcivegaf 4 years ago committed by GitHub
parent 1c1ebd765d
commit 6d97f09195
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 4
      contracts/utils/math/Math.sol
  2. 5
      test/utils/math/Math.test.js

@ -25,8 +25,8 @@ library Math {
* zero.
*/
function average(uint256 a, uint256 b) internal pure returns (uint256) {
// (a + b) / 2 can overflow, so we distribute.
return (a / 2) + (b / 2) + (((a % 2) + (b % 2)) / 2);
// (a + b) / 2 can overflow.
return (a & b) + (a ^ b) / 2;
}
/**

@ -54,6 +54,11 @@ contract('Math', function (accounts) {
const b = new BN('84346');
expect(await this.math.average(a, b)).to.be.bignumber.equal(bnAverage(a, b));
});
it('is correctly calculated with two max uint256 numbers', async function () {
const a = MAX_UINT256;
expect(await this.math.average(a, a)).to.be.bignumber.equal(bnAverage(a, a));
});
});
describe('ceilDiv', function () {

Loading…
Cancel
Save