from ieee754.fpdiv.pipeline import (FPDIVMuxInOut,)
from ieee754.fpcommon.test.fpmux import runfp
+import unittest
from sfpy import Float64, Float32, Float16
+
def sqrt(x):
return x.sqrt()
-def test_pipe_sqrt_fp16():
- dut = FPDIVMuxInOut(16, 4)
- runfp(dut, 16, "test_fpsqrt_pipe_fp16", Float16, sqrt,
- single_op=True, opcode=1, n_vals=100)
-def test_pipe_sqrt_fp32():
- dut = FPDIVMuxInOut(32, 4)
- runfp(dut, 32, "test_fpsqrt_pipe_fp32", Float32, sqrt,
- single_op=True, opcode=1, n_vals=100)
+class TestDivPipe(unittest.TestCase):
+ def test_pipe_sqrt_fp16(self):
+ dut = FPDIVMuxInOut(16, 4)
+ runfp(dut, 16, "test_fpsqrt_pipe_fp16", Float16, sqrt,
+ single_op=True, opcode=1, n_vals=100)
+
+ def test_pipe_sqrt_fp32(self):
+ dut = FPDIVMuxInOut(32, 4)
+ runfp(dut, 32, "test_fpsqrt_pipe_fp32", Float32, sqrt,
+ single_op=True, opcode=1, n_vals=100)
+
+ def test_pipe_sqrt_fp64(self):
+ dut = FPDIVMuxInOut(64, 4)
+ runfp(dut, 64, "test_fpsqrt_pipe_fp64", Float64, sqrt,
+ single_op=True, opcode=1, n_vals=100)
-def test_pipe_sqrt_fp64():
- dut = FPDIVMuxInOut(64, 4)
- runfp(dut, 64, "test_fpsqrt_pipe_fp64", Float64, sqrt,
- single_op=True, opcode=1, n_vals=100)
if __name__ == '__main__':
- test_pipe_sqrt_fp16()
- test_pipe_sqrt_fp32()
- test_pipe_sqrt_fp64()
+ unittest.main()