Add BigInt() to Uint struct in types/uint.go, update tests to check i…

…dent (cosmos#5325)
filippetroviccc · Nov 19, 2019 · c89dee5 · c89dee5
1 parent bb221a5
commit c89dee5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/types/uint.go b/types/uint.go
@@ -13,6 +13,11 @@ type Uint struct {
 	i *big.Int
 }
 
+// BigInt converts Uint to big.Int
+func (u Uint) BigInt() *big.Int {
+	return new(big.Int).Set(u.i)
+}
+
 // NewUintFromBigUint constructs Uint from big.Uint
 func NewUintFromBigInt(i *big.Int) Uint {
 	u, err := checkNewUint(i)

diff --git a/types/uint_test.go b/types/uint_test.go
@@ -84,6 +84,7 @@ func TestIdentUint(t *testing.T) {
 
 		cases := []uint64{
 			i.Uint64(),
+			i.BigInt().Uint64(),
 			i.i.Uint64(),
 			ifromstr.Uint64(),
 			NewUintFromBigInt(new(big.Int).SetUint64(n)).Uint64(),