test string pass/return

Author: Ehsan Totoni

hpat/tests/test_strings.py

@@ -0,0 +1,17 @@
+import unittest
+import hpat
+
+class TestString(unittest.TestCase):
+    def test_pass_return(self):
+        def test_impl(_str):
+            return _str
+        hpat_func = hpat.jit(test_impl)
+        # pass single string and return
+        arg = 'test_str'
+        self.assertEqual(hpat_func(arg), test_impl(arg))
+        # pass string list and return
+        arg = ['test_str1', 'test_str2']
+        self.assertEqual(hpat_func(arg), test_impl(arg))
+
+if __name__ == "__main__":
+    unittest.main()