def _execute_arrow(self):
if isinstance(self._op, FrameNode):
if self._partitions.size == 0:
return pyarrow.Table()
else:
assert self._partitions.size == 1
return self._partitions[0][0].get()
elif isinstance(self._op, MaskNode):
return self._op.input[0]._arrow_row_slice(self._op.row_numeric_idx)
elif isinstance(self._op, TransformNode):
return self._op.input[0]._arrow_col_slice(
set(self._op.exprs.keys()))
elif isinstance(self._op, UnionNode):
return self._arrow_concat(self._op.input)
else:
raise RuntimeError(
f"Unexpected op ({type(self._op)}) in _execute_arrow")
def test_table_ctor_errors():
with pytest.raises(ReferenceError):
repr(pa.Table())
with pytest.raises(ReferenceError):
str(pa.Table())
