src/ieee754/fpdiv/test/test_fpsqrt_pipe_16.py

   1 """ test of FPDIVMuxInOut
   2 """
   3
   4 from ieee754.fpdiv.pipeline import FPDIVMuxInOut
   5 from ieee754.fpcommon.test.case_gen import run_pipe_fp
   6 from ieee754.fpcommon.test import unit_test_half
   7 from ieee754.fpdiv.test.sqrt_data16 import regressions
   8 from ieee754.div_rem_sqrt_rsqrt.core import DivPipeCoreOperation
   9
  10 import unittest
  11 from sfpy import Float16
  12
  13
  14 def sqrt(x):
  15     return x.sqrt()
  16
  17
  18 class TestDivPipe(unittest.TestCase):
  19     def test_pipe_sqrt_fp16(self):
  20         dut = FPDIVMuxInOut(16, 4)
  21         # don't forget to initialize opcode; don't use magic numbers
  22         opcode = int(DivPipeCoreOperation.SqrtRem)
  23         run_pipe_fp(dut, 16, "sqrt16", unit_test_half, Float16, regressions,
  24                     sqrt, 100, single_op=True, opcode=opcode)
  25
  26
  27 if __name__ == '__main__':
  28     unittest.main()