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:
@@ -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]):
|
||||
|
||||
Reference in New Issue
Block a user