src/ieee754/fpadd/test/test_fpadd_pipe.py

   1 """ test of FPADDMuxInOut
   2 """
   3
   4 from ieee754.fpadd.pipeline import (FPADDMuxInOut,)
   5 from ieee754.fpcommon.test.fpmux import runfp
   6
   7 from sfpy import Float64, Float32, Float16
   8 from operator import add
   9
  10 def test_pipe_fp16():
  11     dut = FPADDMuxInOut(16, 4)
  12     runfp(dut, 16, "test_fpadd_pipe_fp16", Float16, add)
  13
  14 def test_pipe_fp32():
  15     dut = FPADDMuxInOut(32, 4)
  16     runfp(dut, 32, "test_fpadd_pipe_fp32", Float32, add)
  17
  18 def test_pipe_fp64():
  19     dut = FPADDMuxInOut(64, 4)
  20     runfp(dut, 64, "test_fpadd_pipe_fp64", Float64, add)
  21
  22 if __name__ == '__main__':
  23     test_pipe_fp16()
  24     test_pipe_fp32()
  25     test_pipe_fp64()
  26