aboutsummaryrefslogtreecommitdiffstats
path: root/src/test_alu.py
blob: af304433061b9e004a0b82f6220b5b8a00e1127b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#!/usr/bin/env python3

from test_utils import *

def make_alu_table() -> str:
  alu = ALU()
  n = 10000
  alu.generate_cases(n)
  alu.process()
  print(f"[test] running {n} ALU test cases")
  return alu.to_table().strip()

class ALU(TestGroup):
  def __init__(self):
    TestGroup.__init__(self, [
      Param("A", 0x0, 0xffff),
      Param("B", 0x0, 0xffff),
      Param("S", 0, 7),
    ], [
      "Y",
      "FLAGS",
    ])

  def process(self):
    new_cases = []
    for kase in self.cases:
      S = kase["S"]
      A = kase["A"]
      B = kase["B"]
      kase["FLAGS"] = "x"

      match S:
        case 0: Y = 0
        case 1: Y = (B - A) & 0xFFFF
        case 2: Y = (A - B) & 0xFFFF
        case 3: Y = (A + B) & 0xFFFF
        case 4: Y = (A ^ B) & 0xFFFF
        case 5: Y = (A | B) & 0xFFFF
        case 6: Y = (A & B) & 0xFFFF
        case 7: Y = 0xffff

      kase["Y"] = Y
      new_cases.append(kase)

    self.cases = new_cases

table = make_alu_table()
run_test("ALU", table)