50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
|
# standard imports
|
||
|
|
import unittest
|
||
|
|
import datetime
|
||
|
|
|
||
|
|
# external imports
|
||
|
|
from chainlib.stat import ChainStat
|
||
|
|
from chainlib.eth.block import Block
|
||
|
|
|
||
|
|
|
||
|
|
class TestStat(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_block(self):
|
||
|
|
|
||
|
|
s = ChainStat()
|
||
|
|
|
||
|
|
d = datetime.datetime.utcnow() - datetime.timedelta(seconds=30)
|
||
|
|
block_a = Block({
|
||
|
|
'timestamp': d.timestamp(),
|
||
|
|
'hash': None,
|
||
|
|
'transactions': [],
|
||
|
|
'number': 41,
|
||
|
|
})
|
||
|
|
|
||
|
|
d = datetime.datetime.utcnow()
|
||
|
|
block_b = Block({
|
||
|
|
'timestamp': d.timestamp(),
|
||
|
|
'hash': None,
|
||
|
|
'transactions': [],
|
||
|
|
'number': 42,
|
||
|
|
})
|
||
|
|
|
||
|
|
s.block_apply(block_a)
|
||
|
|
s.block_apply(block_b)
|
||
|
|
self.assertEqual(s.block_average(), 30.0)
|
||
|
|
|
||
|
|
d = datetime.datetime.utcnow() + datetime.timedelta(seconds=10)
|
||
|
|
block_c = Block({
|
||
|
|
'timestamp': d.timestamp(),
|
||
|
|
'hash': None,
|
||
|
|
'transactions': [],
|
||
|
|
'number': 43,
|
||
|
|
})
|
||
|
|
|
||
|
|
s.block_apply(block_c)
|
||
|
|
self.assertEqual(s.block_average(), 20.0)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|