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]