144 lines
5.6 KiB
Python
Executable file
144 lines
5.6 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
"""
|
|
test.modtest_d.amt: CoinAmt unit tests for the MMGen suite
|
|
"""
|
|
|
|
from decimal import Decimal
|
|
|
|
from mmgen.protocol import init_proto
|
|
from mmgen.tx.new import parse_fee_spec
|
|
from mmgen.cfg import Config
|
|
from mmgen.amt import TokenAmt
|
|
|
|
from ..include.common import cfg, vmsg
|
|
|
|
def get_protos(data):
|
|
return {coin: init_proto(cfg, coin, need_amt=True) for coin in set(d[0] for d in data)}
|
|
|
|
def test_to_unit(data):
|
|
protos = get_protos(data)
|
|
for proto, amt, unit, chk in data:
|
|
amt = protos[proto].coin_amt(amt)
|
|
res = amt.to_unit(unit)
|
|
vmsg(f' {proto.upper()} {amt.fmt(4):<24} => {res:<21} {unit}')
|
|
if '.' in chk:
|
|
assert res == Decimal(chk), f'{res} != {Decimal(chk)}'
|
|
else:
|
|
assert res == int(chk), f'{res} != {int(chk)}'
|
|
return True
|
|
|
|
def test_fee_spec(data):
|
|
protos = get_protos(data)
|
|
for proto, spec, amt, unit in data:
|
|
vmsg(f' {proto.upper():6} {spec:<5} => {amt:<4} {unit}')
|
|
res = parse_fee_spec(protos[proto], spec)
|
|
assert res.amt == amt, f' {res.amt} != {amt}'
|
|
assert res.unit == unit, f' {res.unit} != {unit}'
|
|
return True
|
|
|
|
def test_fmt(data):
|
|
protos = get_protos(data)
|
|
fs = ' {:5} {:18} {:<6} {:<5} {}'
|
|
vmsg(fs.format('PROTO', 'INPUT', 'IWIDTH', 'PREC', 'FORMATTED'))
|
|
for proto, amt, iwidth, prec, chk in data:
|
|
amt = protos[proto].coin_amt(amt)
|
|
args = (iwidth,) if iwidth else ()
|
|
kwargs = {'prec': prec} if prec else {}
|
|
res = amt.fmt(*args, **kwargs)
|
|
vmsg(fs.format(
|
|
proto.upper(),
|
|
f'[{str(amt)}]',
|
|
'None' if iwidth is None else iwidth,
|
|
'None' if prec is None else prec,
|
|
f'[{res}]'))
|
|
assert res == chk, f'[{res}] != [{chk}]'
|
|
return True
|
|
|
|
class unit_tests:
|
|
|
|
altcoin_deps = ('fee_spec_alt', 'to_unit_alt', 'token_amt')
|
|
|
|
def to_unit(self, name, ut, desc='CoinAmt.to_unit() (BTC)'):
|
|
return test_to_unit((
|
|
('btc', '0.00000001', 'satoshi', '1'),
|
|
('btc', '1.23456789', 'satoshi', '123456789')))
|
|
|
|
def to_unit_alt(self, name, ut, desc='CoinAmt.to_unit() (LTC, BCH, ETH, XMR)'):
|
|
return test_to_unit((
|
|
('ltc', '0.00000001', 'satoshi', '1'),
|
|
('ltc', '1.23456789', 'satoshi', '123456789'),
|
|
('bch', '0.00000001', 'satoshi', '1'),
|
|
('bch', '1.23456789', 'satoshi', '123456789'),
|
|
('eth', '1.234567890123456789', 'wei', '1234567890123456789'),
|
|
('eth', '1.234567890123456789', 'Kwei', '1234567890123456.789'),
|
|
('eth', '1.234567890123456789', 'Mwei', '1234567890123.456789'),
|
|
('eth', '1.234567890123456789', 'finney', '1234.567890123456789'),
|
|
('eth', '0.000000012345678901', 'Mwei', '12345.678901'),
|
|
('eth', '0.000000000000000001', 'Kwei', '0.001'),
|
|
('eth', '0.000000000000000001', 'Gwei', '0.000000001'),
|
|
('eth', '0.00000001', 'Gwei', '10'),
|
|
('eth', '1', 'Gwei', '1000000000'),
|
|
('eth', '1', 'finney', '1000'),
|
|
('xmr', '1', 'atomic', '1000000000000'),
|
|
('xmr', '0.000000000001', 'atomic', '1'),
|
|
('xmr', '1.234567890123', 'atomic', '1234567890123')))
|
|
|
|
def fee_spec(self, name, ut, desc='fee spec parsing (BTC)'):
|
|
return test_fee_spec((
|
|
('btc', '32s', '32', 'satoshi'),
|
|
('btc', '1s', '1', 'satoshi')))
|
|
|
|
def fee_spec_alt(self, name, ut, desc='fee spec parsing (LTC, BCH, ETH, XMR)'):
|
|
return test_fee_spec((
|
|
('ltc', '3.07s', '3.07', 'satoshi'),
|
|
('bch', '3.07s', '3.07', 'satoshi'),
|
|
('eth', '3.07G', '3.07', 'Gwei'),
|
|
('eth', '37M', '37', 'Mwei'),
|
|
('eth', '3701w', '3701', 'wei'),
|
|
('eth', '3.07M', '3.07', 'Mwei'),
|
|
('xmr', '3.07a', '3.07', 'atomic')))
|
|
|
|
def fmt(self, name, ut, desc='column formatting (LTC, BCH, ETH, XMR)'):
|
|
return test_fmt((
|
|
('btc', '1', None, None, '1 '),
|
|
('btc', '1.2', None, None, '1.2 '),
|
|
('btc', '1', 1, None, '1 '),
|
|
('btc', '1.2', 1, None, '1.2 '),
|
|
('btc', '12', None, None, '12 '),
|
|
('btc', '12.3', None, None, '12.3 '),
|
|
('btc', '12', 1, None, '12 '),
|
|
('btc', '12.3', 1, None, '12.3 '),
|
|
('btc', '12', 2, None, '12 '),
|
|
('btc', '12.3', 2, None, '12.3 '),
|
|
('btc', '12', 3, None, ' 12 '),
|
|
('btc', '12.3', 3, None, ' 12.3 '),
|
|
('ltc', '0', None, None, '0 '),
|
|
('ltc', '0.00000001', None, None, '0.00000001'),
|
|
('ltc', '0.00000001', 8, None, ' 0.00000001'),
|
|
('xmr', '1.234567890123', None, None, '1.234567890123'),
|
|
('xmr', '1', None, None, '1 '),
|
|
('xmr', '1', None, 4, '1 '),
|
|
('xmr', '123.456', None, None, '123.456 '),
|
|
('xmr', '123.456', 2, None, '123.456 '),
|
|
('xmr', '123.456', None, 4, '123.456 '),
|
|
('xmr', '123.456', 2, 4, '123.456 '),
|
|
('xmr', '1.234567890123', None, 4, '1.2345'),
|
|
('xmr', '1.234567890123', 8, 4, ' 1.2345'),
|
|
('xmr', '1', 8, 4, ' 1 '),
|
|
('xmr', '111', 8, 4, ' 111 ')))
|
|
|
|
def token_amt(self, name, ut, desc='TokenAmt (ETH)'):
|
|
for n, dec, unit, chk in (
|
|
(1234567, 6, 'atomic', '1.234567'),
|
|
('1.234567', 6, None, '1.234567'),
|
|
(1234567, 22, 'atomic', '0.0000000000000001234567'),
|
|
('0.0000000000000001234567', 22, None, '0.0000000000000001234567'),
|
|
):
|
|
amt = TokenAmt(n, decimals=dec, from_unit=unit)
|
|
amt_disp = amt.hl(color=False)
|
|
vmsg(' ' + amt_disp)
|
|
if unit == 'atomic':
|
|
assert amt.to_unit('atomic') == n
|
|
assert amt_disp == chk, f'{amt_disp} != {chk}'
|
|
return True
|