mirror of
git://holbrook.no/erc20-demurrage-token
synced 2024-11-05 02:06:45 +01:00
Add benchmark for gas usage
This commit is contained in:
parent
4dbbf2c9bc
commit
c25e018cd1
167
python/tests/bench_gas.py
Normal file
167
python/tests/bench_gas.py
Normal file
@ -0,0 +1,167 @@
|
||||
# standard imports
|
||||
import os
|
||||
import unittest
|
||||
import json
|
||||
import logging
|
||||
import datetime
|
||||
|
||||
# external imports
|
||||
from chainlib.eth.constant import ZERO_ADDRESS
|
||||
from chainlib.eth.nonce import RPCNonceOracle
|
||||
from chainlib.eth.tx import receipt
|
||||
from chainlib.eth.block import (
|
||||
block_latest,
|
||||
block_by_number,
|
||||
)
|
||||
|
||||
# local imports
|
||||
from erc20_demurrage_token import DemurrageToken
|
||||
|
||||
# test imports
|
||||
from erc20_demurrage_token.unittest import TestDemurrageDefault
|
||||
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
logg = logging.getLogger()
|
||||
|
||||
testdir = os.path.dirname(__file__)
|
||||
|
||||
|
||||
|
||||
class BenchBasic(TestDemurrageDefault):
|
||||
|
||||
def setUp(self):
|
||||
super(BenchBasic, self).setUp()
|
||||
self.bench = {
|
||||
'mint': None,
|
||||
'transfer_light': None,
|
||||
'transfer_heavy': None,
|
||||
'approve': None,
|
||||
'transfer_from': None,
|
||||
'period_light': None,
|
||||
'period_heavy': None,
|
||||
'period_catchup': None,
|
||||
'demurrage': None,
|
||||
}
|
||||
|
||||
|
||||
def test_bench_min(self):
|
||||
nonce_oracle = RPCNonceOracle(self.accounts[0], self.rpc)
|
||||
c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||||
(tx_hash, o) = c.mint_to(self.address, self.accounts[0], self.accounts[1], 1024)
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
self.bench['mint'] = r['gas_used']
|
||||
|
||||
nonce_oracle = RPCNonceOracle(self.accounts[1], self.rpc)
|
||||
c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||||
(tx_hash, o) = c.transfer(self.address, self.accounts[1], self.accounts[2], 512)
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
self.bench['transfer_light'] = r['gas_used']
|
||||
|
||||
(tx_hash, o) = c.approve(self.address, self.accounts[1], self.accounts[0], 512)
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
self.bench['approve'] = r['gas_used']
|
||||
|
||||
nonce_oracle = RPCNonceOracle(self.accounts[0], self.rpc)
|
||||
c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||||
(tx_hash, o) = c.transfer_from(self.address, self.accounts[0], self.accounts[1], self.accounts[3], 256)
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
self.bench['transfer_from'] = r['gas_used']
|
||||
|
||||
z = 0
|
||||
for i in range(100):
|
||||
self.backend.time_travel(self.start_time + int(self.period_seconds / 2) + (10 * (i * (i + 1))))
|
||||
(tx_hash, o) = c.apply_demurrage(self.address, self.accounts[0])
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
z += r['gas_used']
|
||||
logg.info('demurrage round {} gas {}'.format(i, r['gas_used']))
|
||||
z /= 100
|
||||
self.bench['demurrage'] = int(z)
|
||||
|
||||
z = 0
|
||||
for i in range(100):
|
||||
self.backend.time_travel(self.start_time + (self.period_seconds * (i + 1)))
|
||||
(tx_hash, o) = c.change_period(self.address, self.accounts[0])
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
z += r['gas_used']
|
||||
logg.info('period with demurrage round {} gas {}'.format(i, r['gas_used']))
|
||||
|
||||
z /= 100
|
||||
self.bench['period_heavy'] = int(z)
|
||||
|
||||
z = 0
|
||||
for i in range(100):
|
||||
self.backend.time_travel(self.start_time + (self.period_seconds * ((i + 101))))
|
||||
(tx_hash, o) = c.apply_demurrage(self.address, self.accounts[0])
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
|
||||
(tx_hash, o) = c.change_period(self.address, self.accounts[0])
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
z += r['gas_used']
|
||||
logg.info('period without demurrage round {} gas {}'.format(i, r['gas_used']))
|
||||
|
||||
z /= 100
|
||||
self.bench['period_light'] = int(z)
|
||||
|
||||
z = 0
|
||||
self.backend.time_travel(self.start_time + (self.period_seconds * 401))
|
||||
for i in range(100):
|
||||
(tx_hash, o) = c.change_period(self.address, self.accounts[0])
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
z += r['gas_used']
|
||||
logg.info('period catchup round {} gas {}'.format(i, r['gas_used']))
|
||||
|
||||
z /= 100
|
||||
self.bench['period_catchup'] = int(z)
|
||||
|
||||
self.backend.time_travel(self.start_time + (self.period_seconds * 501))
|
||||
(tx_hash, o) = c.mint_to(self.address, self.accounts[0], self.accounts[2], 1024)
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
self.backend.time_travel(self.start_time + (self.period_seconds * 502))
|
||||
|
||||
nonce_oracle = RPCNonceOracle(self.accounts[2], self.rpc)
|
||||
c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle)
|
||||
(tx_hash, o) = c.transfer(self.address, self.accounts[2], self.accounts[4], 1)
|
||||
r = self.rpc.do(o)
|
||||
o = receipt(tx_hash)
|
||||
r = self.rpc.do(o)
|
||||
self.assertEqual(r['status'], 1)
|
||||
self.bench['transfer_heavy'] = r['gas_used']
|
||||
|
||||
print(json.dumps(self.bench))
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
Loading…
Reference in New Issue
Block a user