Improve testing for python 3.9

Author: sergisiso

src/psyclone/tests/psyir/transformations/transformations_test.py

@@ -40,6 +40,7 @@
 API-agnostic tests for various transformation classes.
 '''
 
+import sys
 import os
 import pytest
 from fparser.common.readfortran import FortranStringReader
@@ -590,13 +591,16 @@ def test_omploop_trans_new_options(sample_psyir):
 
     with pytest.raises(TypeError) as excinfo:
         omplooptrans.apply(tree.walk(Loop)[0], collapse="x")
-    # The exact error message depends on the python version, so we only
-    # check it produced the TypeError but not the specific message.
-    # assert ("'OMPLoopTrans' received options with the wrong types:\n"
-    #         "'collapse' option expects type 'int | bool' but "
-    #         "received 'x' of type 'str'.\n"
-    #         "Please see the documentation and check the provided types."
-    #         in str(excinfo.value))
+    if sys.version_info >= (3, 10):
+        assert ("'OMPLoopTrans' received options with the wrong types:\n"
+                "'collapse' option expects type 'int | bool' but "
+                "received 'x' of type 'str'.\n"
+                "Please see the documentation and check the provided types."
+                in str(excinfo.value))
+    else:
+        assert ("The 'collapse' argument must be an integer or a bool but got"
+                " an object of type <class 'str'>"
+                in str(excinfo.value))
 
 
 def test_omplooptrans_apply_nowait(fortran_reader, fortran_writer):