[feature] Add method to get bytecode from a WasmFileHandler (#1235)

pytket/docs/changelog.rst

@@ -1,6 +1,13 @@
 Changelog
 =========
 
+Unreleased
+----------
+
+Features:
+
+* Add ``WasmFileHandler.bytecode()`` method to retrieve the WASM as bytecode.
+
 1.24.0 (January 2024)
 ---------------------
 

pytket/pytket/wasm/wasm.py

@@ -91,7 +91,6 @@ def __init__(self, filepath: str, check_file: bool = True, int_size: int = 32):
         self._unsupported_function = []
 
         if self._check_file:
-
             mod_iter = iter(decode_module(self._wasm_file))
             _, _ = next(mod_iter)
 
@@ -143,7 +142,6 @@ def __init__(self, filepath: str, check_file: bool = True, int_size: int = 32):
                     self._function_types = cur_sec_data.payload.types
 
             for x in function_names:
-
                 # check for only integer type in parameters and return values
                 supported_function = True
                 idx = _func_lookup[x][1]
@@ -214,6 +212,10 @@ def __repr__(self) -> str:
 wasm file was checked"""
             )
 
+    def bytecode(self) -> bytes:
+        """The WASM content as bytecode"""
+        return self._wasm_file
+
     def check_function(
         self, function_name: str, number_of_parameters: int, number_of_returns: int
     ) -> bool:

pytket/tests/classical_test.py

@@ -178,6 +178,10 @@ def test_wasm_2() -> None:
 def test_wasm_3() -> None:
     w = wasm.WasmFileHandler("testfile.wasm")
 
+    with open("testfile.wasm", "rb") as f:
+        bytecode = f.read()
+    assert w.bytecode() == bytecode
+
     c = Circuit(0, 6)
 
     c.add_wasm("add_one", w, [1], [1], [Bit(0), Bit(1)])