From 59b80c75e8b889bf3f8b3bd4561d8be239b92d29 Mon Sep 17 00:00:00 2001 From: Kartik Singhal Date: Wed, 9 Oct 2024 12:28:56 -0500 Subject: [PATCH] test: add couples tests for WasmModuleHandler --- pytket/tests/classical_test.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/pytket/tests/classical_test.py b/pytket/tests/classical_test.py index ceeda6b421..e365074270 100644 --- a/pytket/tests/classical_test.py +++ b/pytket/tests/classical_test.py @@ -192,6 +192,18 @@ def test_wasm_3() -> None: assert c.depth() == 1 +def test_wasm_3_bytes() -> None: + with open("testfile.wasm", "rb") as f: + bytecode = f.read() + c = Circuit(0, 6) + + w = wasm.WasmModuleHandler(bytecode) + + c.add_wasm("add_one", w, [1], [1], [Bit(0), Bit(1)]) + + assert c.depth() == 1 + + def test_wasm_4() -> None: w = wasm.WasmFileHandler("testfile.wasm") @@ -412,6 +424,19 @@ def test_wasm_function_check_6() -> None: assert c.depth() == 1 +def test_wasm_function_check_6_bytes() -> None: + with open("testfile.wasm", "rb") as f: + bytecode = f.read() + + w = wasm.WasmModuleHandler(bytecode) + c = Circuit(20, 20) + c0 = c.add_c_register("c0", 32) + c1 = c.add_c_register("c1", 4) + + c.add_wasm_to_reg("add_one", w, [c0], [c1]) + assert c.depth() == 1 + + def test_wasm_function_check_7() -> None: w = wasm.WasmFileHandler("testfile.wasm", int_size=32) c = Circuit(20, 20)