Add opcode support for divw

pyteal/__init__.pyi

@@ -78,6 +78,7 @@ __all__ = [
     "Div",
     "Mod",
     "Exp",
+    "Divw",
     "BitwiseAnd",
     "BitwiseOr",
     "BitwiseXor",

pyteal/ast/__init__.py

@@ -92,7 +92,7 @@
 )
 
 # ternary ops
-from .ternaryexpr import Ed25519Verify, SetBit, SetByte
+from .ternaryexpr import Divw, Ed25519Verify, SetBit, SetByte
 from .substring import Substring, Extract, Suffix
 
 # more ops
@@ -186,6 +186,7 @@
     "Div",
     "Mod",
     "Exp",
+    "Divw",
     "BitwiseAnd",
     "BitwiseOr",
     "BitwiseXor",

pyteal/ast/ternaryexpr.py

@@ -128,3 +128,25 @@ def SetByte(value: Expr, index: Expr, newByteValue: Expr) -> TernaryExpr:
         index,
         newByteValue,
     )
+
+
+def Divw(hi: Expr, lo: Expr, y: Expr) -> TernaryExpr:
+    """
+    Performs wide division by interpreting `hi` and `lo` as a uint128 value.
+
+    Requires TEAL version 6 or higher.
+
+    Args:
+        hi: Quotient's high 64 bits.  Must evaluate to uint64.
+        lo: Quotient's low 64 bits.  Must evaluate to uint64.
+        y: Divisor.  Must evaluate to uint64.
+
+    """
+    return TernaryExpr(
+        Op.divw,
+        (TealType.uint64, TealType.uint64, TealType.uint64),
+        TealType.uint64,
+        hi,
+        lo,
+        y,
+    )

pyteal/ast/ternaryexpr_test.py

@@ -9,6 +9,7 @@
 teal3Options = CompileOptions(version=3)
 teal4Options = CompileOptions(version=4)
 teal5Options = CompileOptions(version=5)
+teal6Options = CompileOptions(version=6)
 
 
 def test_ed25519verify():
@@ -138,3 +139,35 @@ def test_set_byte_invalid():
 
     with pytest.raises(TealTypeError):
         SetByte(Bytes("base16", "0xFF"), Int(0), Bytes("one"))
+
+
+def test_divw():
+    args = [Int(0), Int(90), Int(30)]
+    expr = Divw(args[0], args[1], args[2])
+    assert expr.type_of() == TealType.uint64
+
+    expected = TealSimpleBlock(
+        [
+            TealOp(args[0], Op.int, args[0].value),
+            TealOp(args[1], Op.int, args[1].value),
+            TealOp(args[2], Op.int, args[2].value),
+            TealOp(expr, Op.divw),
+        ]
+    )
+
+    actual, _ = expr.__teal__(teal6Options)
+    actual.addIncoming()
+    actual = TealBlock.NormalizeBlocks(actual)
+
+    assert actual == expected
+
+
+def test_divw_invalid():
+    with pytest.raises(TealTypeError):
+        Divw(Bytes("10"), Int(0), Int(1))
+
+    with pytest.raises(TealTypeError):
+        Divw(Int(10), Bytes("0"), Int(1))
+
+    with pytest.raises(TealTypeError):
+        Divw(Int(10), Int(0), Bytes("1"))