pass in count and num loops to unit test to make it shorter

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

diff --git a/src/ieee754/fpcommon/test/unit_test_double.py b/src/ieee754/fpcommon/test/unit_test_double.py
index 8cc097f2a2fe1ebdad669d0bdd531d92d796d85a..feed8ffa7a0909719425e7fcb91c4ca81609858e 100644 (file)
--- a/src/ieee754/fpcommon/test/unit_test_double.py
+++ b/src/ieee754/fpcommon/test/unit_test_double.py
@@ -143,85 +143,85 @@ def run_corner_cases(dut, count, op):
     print (count, "vectors passed")
 
 
-def run_edge_cases(dut, count, op):
+def run_edge_cases(dut, count, op, maxcount=1000, num_loops=1000):
     #edge cases
-    stimulus_a = [0x8000000000000000 for i in range(1000)]
-    stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_a = [0x8000000000000000 for i in range(maxcount)]
+    stimulus_b = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_a = [0x0000000000000000 for i in range(1000)]
-    stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_a = [0x0000000000000000 for i in range(maxcount)]
+    stimulus_b = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_b = [0x8000000000000000 for i in range(1000)]
-    stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_b = [0x8000000000000000 for i in range(maxcount)]
+    stimulus_a = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_b = [0x0000000000000000 for i in range(1000)]
-    stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_b = [0x0000000000000000 for i in range(maxcount)]
+    stimulus_a = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_a = [0x7FF8000000000000 for i in range(1000)]
-    stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_a = [0x7FF8000000000000 for i in range(maxcount)]
+    stimulus_b = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_a = [0xFFF8000000000000 for i in range(1000)]
-    stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_a = [0xFFF8000000000000 for i in range(maxcount)]
+    stimulus_b = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_b = [0x7FF8000000000000 for i in range(1000)]
-    stimulus_a = [randint(0, 1<<64) for i in range(1000)]
+    stimulus_b = [0x7FF8000000000000 for i in range(maxcount)]
+    stimulus_a = [randint(0, 1<<64) for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_b = [0xFFF8000000000000 for i in range(1000)]
-    stimulus_a = [randint(0, 1<<64) for i in range(1000)]
+    stimulus_b = [0xFFF8000000000000 for i in range(maxcount)]
+    stimulus_a = [randint(0, 1<<64) for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_a = [0x7FF0000000000000 for i in range(1000)]
-    stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_a = [0x7FF0000000000000 for i in range(maxcount)]
+    stimulus_b = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_a = [0xFFF0000000000000 for i in range(1000)]
-    stimulus_b = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_a = [0xFFF0000000000000 for i in range(maxcount)]
+    stimulus_b = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_b = [0x7FF0000000000000 for i in range(1000)]
-    stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_b = [0x7FF0000000000000 for i in range(maxcount)]
+    stimulus_a = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
-    stimulus_b = [0xFFF0000000000000 for i in range(1000)]
-    stimulus_a = [randint(0, 1<<64)  for i in range(1000)]
+    stimulus_b = [0xFFF0000000000000 for i in range(maxcount)]
+    stimulus_a = [randint(0, 1<<64)  for i in range(maxcount)]
     yield from run_test(dut, stimulus_a, stimulus_b, op)
     count += len(stimulus_a)
     print (count, "vectors passed")
 
     #seed(0)
-    for i in range(100000):
-        stimulus_a = [randint(0, 1<<64) for i in range(1000)]
-        stimulus_b = [randint(0, 1<<64) for i in range(1000)]
+    for i in range(num_loops):
+        stimulus_a = [randint(0, 1<<64) for i in range(maxcount)]
+        stimulus_b = [randint(0, 1<<64) for i in range(maxcount)]
         yield from run_test(dut, stimulus_a, stimulus_b, op)
-        count += 1000
+        count += maxcount
         print (count, "random vectors passed")