forked from LBRYCommunity/lbry-sdk
96 lines
3 KiB
Python
96 lines
3 KiB
Python
import unittest
|
|
|
|
from lbry.wallet.util import ArithUint256
|
|
from lbry.wallet.util import coins_to_satoshis as c2s, satoshis_to_coins as s2c
|
|
|
|
|
|
class TestCoinValueParsing(unittest.TestCase):
|
|
|
|
def test_good_output(self):
|
|
self.assertEqual(s2c(1), "0.00000001")
|
|
self.assertEqual(s2c(10**7), "0.1")
|
|
self.assertEqual(s2c(2*10**8), "2.0")
|
|
self.assertEqual(s2c(2*10**17), "2000000000.0")
|
|
|
|
def test_good_input(self):
|
|
self.assertEqual(c2s("0.00000001"), 1)
|
|
self.assertEqual(c2s("0.1"), 10**7)
|
|
self.assertEqual(c2s("1.0"), 10**8)
|
|
self.assertEqual(c2s("2.00000000"), 2*10**8)
|
|
self.assertEqual(c2s("2000000000.0"), 2*10**17)
|
|
|
|
def test_bad_input(self):
|
|
with self.assertRaises(ValueError):
|
|
c2s("1")
|
|
with self.assertRaises(ValueError):
|
|
c2s("-1.0")
|
|
with self.assertRaises(ValueError):
|
|
c2s("10000000000.0")
|
|
with self.assertRaises(ValueError):
|
|
c2s("1.000000000")
|
|
with self.assertRaises(ValueError):
|
|
c2s("-0")
|
|
with self.assertRaises(ValueError):
|
|
c2s("1")
|
|
with self.assertRaises(ValueError):
|
|
c2s(".1")
|
|
with self.assertRaises(ValueError):
|
|
c2s("1e-7")
|
|
|
|
|
|
class TestArithUint256(unittest.TestCase):
|
|
|
|
def test_arithunit256(self):
|
|
# https://github.com/bitcoin/bitcoin/blob/master/src/test/arith_uint256_tests.cpp
|
|
|
|
from_compact = ArithUint256.from_compact
|
|
eq = self.assertEqual
|
|
|
|
eq(from_compact(0).value, 0)
|
|
eq(from_compact(0x00123456).value, 0)
|
|
eq(from_compact(0x01003456).value, 0)
|
|
eq(from_compact(0x02000056).value, 0)
|
|
eq(from_compact(0x03000000).value, 0)
|
|
eq(from_compact(0x04000000).value, 0)
|
|
eq(from_compact(0x00923456).value, 0)
|
|
eq(from_compact(0x01803456).value, 0)
|
|
eq(from_compact(0x02800056).value, 0)
|
|
eq(from_compact(0x03800000).value, 0)
|
|
eq(from_compact(0x04800000).value, 0)
|
|
|
|
# Make sure that we don't generate compacts with the 0x00800000 bit set
|
|
uint = ArithUint256(0x80)
|
|
eq(uint.compact, 0x02008000)
|
|
|
|
uint = from_compact(0x01123456)
|
|
eq(uint.value, 0x12)
|
|
eq(uint.compact, 0x01120000)
|
|
|
|
uint = from_compact(0x01fedcba)
|
|
eq(uint.value, 0x7e)
|
|
eq(uint.negative, 0x01fe0000)
|
|
|
|
uint = from_compact(0x02123456)
|
|
eq(uint.value, 0x1234)
|
|
eq(uint.compact, 0x02123400)
|
|
|
|
uint = from_compact(0x03123456)
|
|
eq(uint.value, 0x123456)
|
|
eq(uint.compact, 0x03123456)
|
|
|
|
uint = from_compact(0x04123456)
|
|
eq(uint.value, 0x12345600)
|
|
eq(uint.compact, 0x04123456)
|
|
|
|
uint = from_compact(0x04923456)
|
|
eq(uint.value, 0x12345600)
|
|
eq(uint.negative, 0x04923456)
|
|
|
|
uint = from_compact(0x05009234)
|
|
eq(uint.value, 0x92340000)
|
|
eq(uint.compact, 0x05009234)
|
|
|
|
uint = from_compact(0x20123456)
|
|
eq(uint.value, 0x1234560000000000000000000000000000000000000000000000000000000000)
|
|
eq(uint.compact, 0x20123456)
|