lbry-sdk/tests/unit/wallet/test_utils.py

import unittest

from lbry.wallet.client.util import ArithUint256
from lbry.wallet.client.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)