Allow for NULL converted type (#340)

* Allow for NULL converted type * Add test and test file
dask · Jun 5, 2018 · 35a1b0b · 35a1b0b
1 parent fce1cfd
commit 35a1b0b
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 0 deletions.
diff --git a/fastparquet/parquet.thrift b/fastparquet/parquet.thrift
@@ -187,6 +187,7 @@ enum ConvertedType {
    * particular timezone or date.
    */
   INTERVAL = 21;
+  NULL = 24;
 
   /**
    * Annotates a column that is always null

diff --git a/fastparquet/parquet_thrift/parquet/ttypes.py b/fastparquet/parquet_thrift/parquet/ttypes.py
diff --git a/fastparquet/test/test_api.py b/fastparquet/test/test_api.py
@@ -656,3 +656,10 @@ def test_only_partition_columns(tempdir):
     with pytest.raises(ValueError):
         # because this leaves no data to write
         write(tempdir, df[['b']], file_scheme='hive', partition_on=['b'])
+
+
+def test_empty_df():
+    p = ParquetFile(os.path.join(TEST_DATA, "empty.parquet"))
+    df = p.to_pandas()
+    assert list(p.columns) == ['a', 'b', 'c', '__index_level_0__']
+    assert len(df) == 0
diff --git a/test-data/empty.parquet b/test-data/empty.parquet
diff --git a/test-data/test.parquet b/test-data/test.parquet