Tighten int_to_bytes32

cburgdorf · cburgdorf · commit 7d5e5ad62921 · 2018-10-19T17:46:36.000+02:00
diff --git a/eth/utils/numeric.py b/eth/utils/numeric.py
@@ -7,6 +7,9 @@
 from cytoolz import (
     curry,
 )
+from eth_typing import (
+    Hash32,
+)
 
 from eth.constants import (
     UINT_255_MAX,
@@ -15,7 +18,7 @@
 )
 
 
-def int_to_bytes32(value: Union[int, bool]) -> bytes:
+def int_to_bytes32(value: Union[int, bool]) -> Hash32:
     if not isinstance(value, int) or isinstance(value, bool):
         raise ValueError(
             "Value must be an integer: Got: {0}".format(
@@ -35,7 +38,7 @@ def int_to_bytes32(value: Union[int, bool]) -> bytes:
             )
         )
     value_bytes = value.to_bytes(32, 'big')
-    return value_bytes
+    return Hash32(value_bytes)
 
 
 def ceilXX(value: int, ceiling: int) -> int:

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,9 @@`
`7`	`7`	`from cytoolz import (`
`8`	`8`	`curry,`
`9`	`9`	`)`
	`10`	`+from eth_typing import (`
	`11`	`+ Hash32,`
	`12`	`+)`
`10`	`13`
`11`	`14`	`from eth.constants import (`
`12`	`15`	`UINT_255_MAX,`
`@@ -15,7 +18,7 @@`
`15`	`18`	`)`
`16`	`19`
`17`	`20`
`18`		`-def int_to_bytes32(value: Union[int, bool]) -> bytes:`
	`21`	`+def int_to_bytes32(value: Union[int, bool]) -> Hash32:`
`19`	`22`	`if not isinstance(value, int) or isinstance(value, bool):`
`20`	`23`	`raise ValueError(`
`21`	`24`	`"Value must be an integer: Got: {0}".format(`
`@@ -35,7 +38,7 @@ def int_to_bytes32(value: Union[int, bool]) -> bytes:`
`35`	`38`	`)`
`36`	`39`	`)`
`37`	`40`	`value_bytes = value.to_bytes(32, 'big')`
`38`		`- return value_bytes`
	`41`	`+ return Hash32(value_bytes)`
`39`	`42`
`40`	`43`
`41`	`44`	`def ceilXX(value: int, ceiling: int) -> int:`