convert all test_caller*.py to work with pytest/unittest test discovery

[openpower-isa.git] / src / openpower / decoder / isa / test_caller_mul.py

diff --git a/src/openpower/decoder/isa/test_caller_mul.py b/src/openpower/decoder/isa/test_caller_mul.py
index cb2e218de49f8872f2f6380db8ee6bb6478cae04..24b2f00de8fcae49fc2366c24e0aed990048f68f 100644 (file)
--- a/src/openpower/decoder/isa/test_caller_mul.py
+++ b/src/openpower/decoder/isa/test_caller_mul.py
@@ -6,33 +6,21 @@ related bugs:
 """
 
 import unittest
-import sys
-
-# These tests utilize the run_hdl=False parameter to compare
-# simulator with expected states
-from soc.simple.test.test_runner import TestRunner
+from openpower.test.runner import TestRunnerBase
 from openpower.test.mul.mul_cases import MulTestCases2Arg
 
+# writing the test_caller invocation this way makes it work with pytest
 
-if __name__ == "__main__":
-
-    # allow list of testing to be selected by command-line
-    testing = sys.argv[1:]
-    sys.argv = sys.argv[:1]
 
-    if not testing:
-        testing = ['mul2']
+class TestMul2Arg(TestRunnerBase):
+    def __init__(self, test):
+        assert test == 'test'
+        super().__init__(MulTestCases2Arg().test_data)
 
-    unittest.main(exit=False)
-    suite = unittest.TestSuite()
+    def test(self):
+        # dummy function to make unittest try to test this class
+        pass
 
-    # dictionary  of data for tests
-    tests = {'mul2': MulTestCases2Arg().test_data}
 
-    # walk through all tests, those requested get added
-    for tname, data in tests.items():
-        if tname in testing:
-            suite.addTest(TestRunner(data, run_hdl=False))
-
-    runner = unittest.TextTestRunner()
-    runner.run(suite)
+if __name__ == "__main__":
+    unittest.main()