test: add unit tests and allow test files in repo

Author: deaneeth

.gitignore

@@ -1,8 +1,3 @@
-test_assembler.py
-test_gpu_core.py
-test_gpu.py
-test_programs.py
-
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]

tests/test_assembler.py

@@ -0,0 +1,24 @@
+
+import tempfile
+import os
+from tinygpu.assembler import assemble_file
+
+def test_assemble_file_basic():
+	code = """
+	start:
+	LOAD R1, 42
+	ADD R2, R1, 1
+	JMP start
+	"""
+	with tempfile.NamedTemporaryFile("w", delete=False, suffix=".tgpu") as f:
+		f.write(code)
+		fname = f.name
+	try:
+		program, labels = assemble_file(fname)
+		assert labels["start"] == 0
+		assert program[0][0] == "LOAD"
+		assert program[1][0] == "ADD"
+		assert program[2][0] == "JMP"
+		assert program[2][1][0] == "start" or program[2][1][0] == 0
+	finally:
+		os.remove(fname)

tests/test_gpu.py

@@ -0,0 +1,39 @@
+
+import numpy as np
+from tinygpu.gpu import TinyGPU
+
+def test_tinygpu_init():
+	gpu = TinyGPU(num_threads=4, num_registers=8, mem_size=16)
+	assert gpu.num_threads == 4
+	assert gpu.num_registers == 8
+	assert gpu.mem_size == 16
+	assert gpu.registers.shape == (4, 8)
+	assert gpu.memory.shape == (16,)
+	assert gpu.pc.shape == (4,)
+	assert gpu.active.shape == (4,)
+
+def test_tinygpu_step_and_run():
+	gpu = TinyGPU(num_threads=2, num_registers=4, mem_size=8)
+	# Fake program: increment R0, halt after 2 steps
+	gpu.program = [
+		("LOAD", [("R", 0), 1]),
+		("ADD", [("R", 0), ("R", 0), 1]),
+	]
+	gpu.pc[:] = 0
+	gpu.active[:] = True
+	gpu.sync_waiting[:] = False
+	gpu.history_registers = []
+	gpu.history_memory = []
+	gpu.history_pc = []
+	# Patch INSTRUCTIONS for test
+	from tinygpu import instructions
+	def fake_load(self, tid, reg, val):
+		self.registers[tid, reg[1]] = val
+	def fake_add(self, tid, reg, reg2, val):
+		self.registers[tid, reg[1]] = self.registers[tid, reg2[1]] + val
+	instructions.INSTRUCTIONS["LOAD"] = fake_load
+	instructions.INSTRUCTIONS["ADD"] = fake_add
+	# After first step, both instructions are executed for both threads
+	gpu.step()
+	# LOAD sets R0=1, then ADD sets R0=2 in the same step
+	assert np.all(gpu.registers[:,0] == 2)