add test_add fp16

[ieee754fpu.git] / src / ieee754 / fpcommon / test / case_gen.py

diff --git a/src/ieee754/fpcommon/test/case_gen.py b/src/ieee754/fpcommon/test/case_gen.py
index 441b093e45c5f58056a49dae5d1587ac81c0dbd5..5a1661a159c6c1167cd6580e23ad60aa6bfc7457 100644 (file)
--- a/src/ieee754/fpcommon/test/case_gen.py
+++ b/src/ieee754/fpcommon/test/case_gen.py
@@ -25,17 +25,20 @@ def replicate(fixed_num, maxcount):
     else:
         return fixed_num
 
+def get_rval(width):
+    mval = (1<<width)-1
+    return randint(0, mval)
 
 def get_rand1(mod, fixed_num, maxcount, width):
     stimulus_a = replicate(fixed_num, maxcount)
-    stimulus_b = [randint(0, 1<<width) for i in range(maxcount)]
+    stimulus_b = [get_rval(width) for i in range(maxcount)]
     return zip(stimulus_a, stimulus_b)
 
 
 def get_nan_noncan(mod, fixed_num, maxcount, width):
     stimulus_a = replicate(fixed_num, maxcount)
     # non-canonical NaNs.
-    stimulus_b = [mod.set_exponent(randint(0, 1<<width), mod.max_e) \
+    stimulus_b = [mod.set_exponent(get_rval(width), mod.max_e) \
                         for i in range(maxcount)]
     return zip(stimulus_a, stimulus_b)
 
@@ -43,7 +46,7 @@ def get_nan_noncan(mod, fixed_num, maxcount, width):
 def get_n127(mod, fixed_num, maxcount, width):
     stimulus_a = replicate(fixed_num, maxcount)
     # -127
-    stimulus_b = [mod.set_exponent(randint(0, 1<<width), -mod.max_e+1) \
+    stimulus_b = [mod.set_exponent(get_rval(width), -mod.max_e+1) \
                         for i in range(maxcount)]
     return zip(stimulus_a, stimulus_b)
 
@@ -51,7 +54,7 @@ def get_n127(mod, fixed_num, maxcount, width):
 def get_nearly_zero(mod, fixed_num, maxcount, width):
     stimulus_a = replicate(fixed_num, maxcount)
     # nearly zero
-    stimulus_b = [mod.set_exponent(randint(0, 1<<width), -mod.max_e+2) \
+    stimulus_b = [mod.set_exponent(get_rval(width), -mod.max_e+2) \
                         for i in range(maxcount)]
     return zip(stimulus_a, stimulus_b)
 
@@ -59,7 +62,7 @@ def get_nearly_zero(mod, fixed_num, maxcount, width):
 def get_nearly_inf(mod, fixed_num, maxcount, width):
     stimulus_a = replicate(fixed_num, maxcount)
     # nearly inf
-    stimulus_b = [mod.set_exponent(randint(0, 1<<width), mod.max_e-1) \
+    stimulus_b = [mod.set_exponent(get_rval(width), mod.max_e-1) \
                         for i in range(maxcount)]
     return zip(stimulus_a, stimulus_b)
 
@@ -67,7 +70,7 @@ def get_nearly_inf(mod, fixed_num, maxcount, width):
 def get_corner_rand(mod, fixed_num, maxcount, width):
     stimulus_a = replicate(fixed_num, maxcount)
     # random
-    stimulus_b = [randint(0, 1<<width) for i in range(maxcount)]
+    stimulus_b = [get_rval(width) for i in range(maxcount)]
     return zip(stimulus_a, stimulus_b)