def test_head(tempdir): fn = os.path.join(tempdir, 'test.parquet') val = [2, 10, 34, 76] df = pd.DataFrame({'val': val}) write(fn, df) pf = ParquetFile(fn) assert pf.head(1).val.tolist() == [2]
def test_head(tempdir): dn = os.path.join(tempdir, 'test_parquet') val = [2, 10, 34, 76] df = pd.DataFrame({'val': val}) write(dn, df, row_group_offsets=[0, 2], file_scheme='hive') pf = ParquetFile(dn) assert pf.head(3).val.tolist() == [2, 10, 34]