bug 676: put maxloc.py test data into array
authorLuke Kenneth Casson Leighton <lkcl@lkcl.net>
Fri, 15 Dec 2023 23:28:53 +0000 (23:28 +0000)
committerLuke Kenneth Casson Leighton <lkcl@lkcl.net>
Fri, 15 Dec 2023 23:28:53 +0000 (23:28 +0000)
src/openpower/decoder/isa/maxloc.py

index 345dd11c8b1a6c25b67df16a5f1213216444e05a..83d89569785ff00a461d1f32b3d1dc07e93200e5 100644 (file)
@@ -22,27 +22,16 @@ def m2(a):
 
 # /*Testbench*/
 
-if __name__ == '__main__':
-    arr = [5,2,8,1,3,7,9,4]
-    print("search list", arr)
-    result = m2(arr)
-    print("Index of the maximum value in an array is: %d" % result)
-    assert (result == 6)
-
-    arr = [5,2,8,9,9,7,9,4]
-    print("search list", arr)
-    result = m2(arr)
-    print("Index of the maximum value in an array is: %d" % result)
-    assert (result == 3)
+test_data = [
+    ([5,2,8,1,3,7,9,4], 6),
+    ([5,2,8,9,9,7,9,4], 3),
+    ([0,0,0,0,0,0,0,0], 0),
+    ([5,5,5,5,5,5,5,5], 0),
+]
 
-    arr = [0,0,0,0,0,0,0,0]
-    print("search list", arr)
-    result = m2(arr)
-    print("Index of the maximum value in an array is: %d" % result)
-    assert (result == 0)
-    
-    arr = [5,5,5,5,5,5,5,5]
-    print("search list", arr)
-    result = m2(arr)
-    print("Index of the maximum value in an array is: %d" % result)
-    assert (result == 0)
+if __name__ == '__main__':
+    for arr, expected in test_data:
+        print("search list", arr)
+        result = m2(arr)
+        print("Index of the maximum value in an array is: %d" % result)
+        assert (result == expected)