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())