Merge branch 'master' into test-class-not-runnning

Author: max-sixty

docs/source/changelog.rst

@@ -1,6 +1,10 @@
 Changelog
 =========
 
+0.3.2 / [TBD]
+------------------
+- Fix bug with querying for an array of floats (:issue:`123`)
+
 0.3.1 / 2018-02-13
 ------------------
 

pandas_gbq/gbq.py

@@ -720,6 +720,8 @@ def _parse_data(schema, rows):
     col_names = [str(field['name']) for field in fields]
     col_dtypes = [
         dtype_map.get(field['type'].upper(), object)
+        if field['mode'] != 'repeated'
+        else object
         for field in fields
     ]
     page_array = np.zeros((len(rows),), dtype=lzip(col_names, col_dtypes))

pandas_gbq/tests/test_gbq.py

@@ -365,12 +365,16 @@ def test_read_gbq_with_no_project_id_given_should_fail(self):
             gbq.read_gbq('SELECT 1')
 
     def test_that_parse_data_works_properly(self):
+
+        from google.cloud.bigquery.table import Row
         test_schema = {'fields': [
-            {'mode': 'NULLABLE', 'name': 'valid_string', 'type': 'STRING'}]}
-        test_page = [{'f': [{'v': 'PI'}]}]
+            {'mode': 'NULLABLE', 'name': 'column_x', 'type': 'STRING'}]}
+        field_to_index = {'column_x': 0}
+        values = ('row_value',)
+        test_page = [Row(values, field_to_index)]
 
         test_output = gbq._parse_data(test_schema, test_page)
-        correct_output = DataFrame({'valid_string': ['PI']})
+        correct_output = DataFrame({'column_x': ['row_value']})
         tm.assert_frame_equal(test_output, correct_output)
 
     def test_read_gbq_with_invalid_private_key_json_should_fail(self):
@@ -973,6 +977,14 @@ def test_array_agg(self):
         tm.assert_frame_equal(df, DataFrame([["a", [1, 3]], ["b", [2]]],
                               columns=["letter", "numbers"]))
 
+    def test_array_of_floats(self):
+        query = """select [1.1, 2.2, 3.3] as a, 4 as b"""
+        df = gbq.read_gbq(query, project_id=_get_project_id(),
+                          private_key=_get_private_key_path(),
+                          dialect='standard')
+        tm.assert_frame_equal(df, DataFrame([[[1.1, 2.2, 3.3], 4]],
+                              columns=["a", "b"]))
+
 
 class TestToGBQIntegrationWithServiceAccountKeyPath(object):
     # Changes to BigQuery table schema may take up to 2 minutes as of May 2015

setup.py

@@ -29,7 +29,7 @@ def readme():
     name=NAME,
     version=versioneer.get_version(),
     cmdclass=versioneer.get_cmdclass(),
-    description="Pandas interface to Google Big Query",
+    description="Pandas interface to Google BigQuery",
     long_description=readme(),
     license='BSD License',
     author='The PyData Development Team',