schema as property

This commit is contained in:
Krisztián Szűcs committed 2019-12-12 12:27:38 +01:00
1 parent e9f77bd6b8
commit 620ba6ffad
2 files changed
+2 -1

No files matched your search

+1
View File
@@ -967,6 +967,7 @@ cdef class ScannerBuilder:
check_status(self.builder.UseThreads(value))
return self
@property
def schema(self):
return pyarrow_wrap_schema(self.builder.schema())
+1 -1
View File
@@ -214,7 +214,7 @@ def test_dataset(simple_data_source, tree_data_source, schema):
# TODO(kszucs): test non-boolean expressions for filter do raise
builder = dataset.new_scan().use_threads(True).filter(condition)
assert isinstance(builder, ds.ScannerBuilder)
assert isinstance(builder.schema(), pa.Schema)
assert isinstance(builder.schema, pa.Schema)
scanner = builder.finish()
assert isinstance(scanner, ds.Scanner)