From 416e896956d3b05793337da6609fe1544797c0a9 Mon Sep 17 00:00:00 2001
From: Jan Tuomi <jans.tuomi@gmail.com>
Date: Thu, 22 Feb 2024 08:52:20 +0200
Subject: Add stuff, refactor

---
 test/utils.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 test/utils.py

(limited to 'test/utils.py')

diff --git a/test/utils.py b/test/utils.py
new file mode 100644
index 0000000..873aa64
--- /dev/null
+++ b/test/utils.py
@@ -0,0 +1,25 @@
+def pad_bytearray(ba: bytearray, to_length: int = 64 * 1024) -> bytearray:
+  """
+  Pads a bytearray with zeros until it reaches a specified length.
+
+  :param to_length: Target length in bytes.
+  :param ba: The bytearray to be padded.
+  :return: The padded bytearray.
+  """
+  padding_size = to_length - len(ba)
+  if padding_size > 0:
+    ba.extend(b'\x00' * padding_size)
+  return ba
+
+def make_rom(words: list[int]) -> bytearray:
+  """
+  Constructs a ROM image from a list of 16-bit words.
+
+  :param words: The list of 16-bit words.
+  :return: The ROM image as a bytearray.
+  """
+  bytes = []
+  for word in words:
+    bytes.append((word >> 8) & 0xFF)
+    bytes.append(word & 0xFF)
+  return pad_bytearray(bytearray(bytes))
-- 
cgit v1.3