Fix revert column behavior

Now all columns are tested in the parquet test.

Signed-off-by: Will Beason <willbeason@gmail.com>
This commit is contained in:
Will Beason
2025-06-03 15:03:33 -05:00
parent 06a784ef27
commit 123b9a18a8
3 changed files with 25 additions and 13 deletions

View File

@@ -58,13 +58,12 @@ class RevisionTable:
def schema(self) -> pa.Schema:
return pa.schema([c.field for c in self.columns])
def pop(self):
schema = self.schema()
data = []
def pop(self) -> dict:
data = {}
for column in self.columns:
data.append(column.pop())
data[column.field.name] = column.pop()
return pa.table(data, schema)
return data
class RevisionId(RevisionField[int]):