support Global and FreeVar in parquet file

Author: Ehsan Totoni

hpat/io/parquet_pio.py

@@ -90,7 +90,8 @@ def gen_parquet_read(self, file_name, lhs):
 
         if table_types is None:
             fname_def = guard(get_definition, self.func_ir, file_name)
-            if not isinstance(fname_def, ir.Const) or not isinstance(fname_def.value, str):
+            if (not isinstance(fname_def, (ir.Const, ir.Global, ir.FreeVar))
+                    or not isinstance(fname_def.value, str)):
                 raise ValueError("Parquet schema not available")
             file_name_str = fname_def.value
             col_names, col_types = parquet_file_schema(file_name_str)

hpat/tests/test_io.py

@@ -10,6 +10,9 @@
     get_start_end)
 
 
+kde_file = 'kde.parquet'
+
+
 class TestIO(unittest.TestCase):
 
     def setUp(self):
@@ -190,6 +193,29 @@ def test_impl():
         self.assertEqual(count_array_REPs(), 0)
         self.assertEqual(count_parfor_REPs(), 0)
 
+    def test_pq_read_global_str1(self):
+        def test_impl():
+            df = pd.read_parquet(kde_file)
+            X = df['points']
+            return X.sum()
+
+        hpat_func = hpat.jit(test_impl)
+        np.testing.assert_almost_equal(hpat_func(), test_impl())
+        self.assertEqual(count_array_REPs(), 0)
+        self.assertEqual(count_parfor_REPs(), 0)
+
+    def test_pq_read_freevar_str1(self):
+        kde_file2 = 'kde.parquet'
+        def test_impl():
+            df = pd.read_parquet(kde_file2)
+            X = df['points']
+            return X.sum()
+
+        hpat_func = hpat.jit(test_impl)
+        np.testing.assert_almost_equal(hpat_func(), test_impl())
+        self.assertEqual(count_array_REPs(), 0)
+        self.assertEqual(count_parfor_REPs(), 0)
+
     def test_pd_read_parquet(self):
         def test_impl():
             df = pd.read_parquet('kde.parquet')