src/add/test_add64.py

   1 from nmigen import Module, Signal
   2 from nmigen.compat.sim import run_simulation
   3 from operator import add
   4
   5 from nmigen_add_experiment import FPADD
   6
   7 import sys
   8 import atexit
   9 from random import randint
  10 from random import seed
  11
  12 from unit_test_double import (get_mantissa, get_exponent, get_sign, is_nan,
  13                                 is_inf, is_pos_inf, is_neg_inf,
  14                                 match, get_case, check_case, run_test)
  15
  16
  17 def testbench(dut):
  18     yield from check_case(dut, 0, 0, 0)
  19     yield from check_case(dut, 0x3FF0000000000000, 0x4000000000000000,
  20                                0x4008000000000000)
  21     yield from check_case(dut, 0x4000000000000000, 0x3FF0000000000000,
  22                                0x4008000000000000)
  23     yield from check_case(dut, 0x4056C00000000000, 0x4042800000000000,
  24                                0x4060000000000000)
  25     yield from check_case(dut, 0x4056C00000000000, 0x4042EA3D70A3D70A,
  26                                0x40601A8F5C28F5C2)
  27
  28     count = 0
  29
  30     #regression tests
  31     stimulus_a = [0x3ff00000000000c5, 0xff80000000000000]
  32     stimulus_b = [0xbd28a404211fb72b, 0x7f80000000000000]
  33     yield from run_test(dut, stimulus_a, stimulus_b, add)
  34     count += len(stimulus_a)
  35     print (count, "vectors passed")
  36
  37     #corner cases
  38     from itertools import permutations
  39     stimulus_a = [i[0] for i in permutations([
  40         0x8000000000000000,
  41         0x0000000000000000,
  42         0x7ff8000000000000,
  43         0xfff8000000000000,
  44         0x7ff0000000000000,
  45         0xfff0000000000000
  46     ], 2)]
  47     stimulus_b = [i[1] for i in permutations([
  48         0x8000000000000000,
  49         0x0000000000000000,
  50         0x7ff8000000000000,
  51         0xfff8000000000000,
  52         0x7ff0000000000000,
  53         0xfff0000000000000
  54     ], 2)]
  55     yield from run_test(dut, stimulus_a, stimulus_b, add)
  56     count += len(stimulus_a)
  57     print (count, "vectors passed")
  58
  59     #edge cases
  60     stimulus_a = [0x8000000000000000 for i in range(1000)]
  61     stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
  62     yield from run_test(dut, stimulus_a, stimulus_b, add)
  63     count += len(stimulus_a)
  64     print (count, "vectors passed")
  65
  66     stimulus_a = [0x0000000000000000 for i in range(1000)]
  67     stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
  68     yield from run_test(dut, stimulus_a, stimulus_b, add)
  69     count += len(stimulus_a)
  70     print (count, "vectors passed")
  71
  72     stimulus_b = [0x8000000000000000 for i in range(1000)]
  73     stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
  74     yield from run_test(dut, stimulus_a, stimulus_b, add)
  75     count += len(stimulus_a)
  76     print (count, "vectors passed")
  77
  78     stimulus_b = [0x0000000000000000 for i in range(1000)]
  79     stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
  80     yield from run_test(dut, stimulus_a, stimulus_b, add)
  81     count += len(stimulus_a)
  82     print (count, "vectors passed")
  83
  84     stimulus_a = [0x7FF8000000000000 for i in range(1000)]
  85     stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
  86     yield from run_test(dut, stimulus_a, stimulus_b, add)
  87     count += len(stimulus_a)
  88     print (count, "vectors passed")
  89
  90     stimulus_a = [0xFFF8000000000000 for i in range(1000)]
  91     stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
  92     yield from run_test(dut, stimulus_a, stimulus_b, add)
  93     count += len(stimulus_a)
  94     print (count, "vectors passed")
  95
  96     stimulus_b = [0x7FF8000000000000 for i in range(1000)]
  97     stimulus_a = [randint(0, 1<<64) for i in range(1000)]
  98     yield from run_test(dut, stimulus_a, stimulus_b, add)
  99     count += len(stimulus_a)
 100     print (count, "vectors passed")
 101
 102     stimulus_b = [0xFFF8000000000000 for i in range(1000)]
 103     stimulus_a = [randint(0, 1<<64) for i in range(1000)]
 104     yield from run_test(dut, stimulus_a, stimulus_b, add)
 105     count += len(stimulus_a)
 106     print (count, "vectors passed")
 107
 108     stimulus_a = [0x7FF0000000000000 for i in range(1000)]
 109     stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
 110     yield from run_test(dut, stimulus_a, stimulus_b, add)
 111     count += len(stimulus_a)
 112     print (count, "vectors passed")
 113
 114     stimulus_a = [0xFFF0000000000000 for i in range(1000)]
 115     stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
 116     yield from run_test(dut, stimulus_a, stimulus_b, add)
 117     count += len(stimulus_a)
 118     print (count, "vectors passed")
 119
 120     stimulus_b = [0x7FF0000000000000 for i in range(1000)]
 121     stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
 122     yield from run_test(dut, stimulus_a, stimulus_b, add)
 123     count += len(stimulus_a)
 124     print (count, "vectors passed")
 125
 126     stimulus_b = [0xFFF0000000000000 for i in range(1000)]
 127     stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
 128     yield from run_test(dut, stimulus_a, stimulus_b, add)
 129     count += len(stimulus_a)
 130     print (count, "vectors passed")
 131
 132     #seed(0)
 133     for i in range(100000):
 134         stimulus_a = [randint(0, 1<<64) for i in range(1000)]
 135         stimulus_b = [randint(0, 1<<64) for i in range(1000)]
 136         yield from run_test(dut, stimulus_a, stimulus_b, add)
 137         count += 1000
 138         print (count, "random vectors passed")
 139
 140
 141 if __name__ == '__main__':
 142     dut = FPADD(width=64, single_cycle=True)
 143     run_simulation(dut, testbench(dut), vcd_name="test_add64.vcd")
 144