def AsROOTFile(self, columns=[]):
r"""
Terminal - take the AST and return a root file.
columns - Array of names of the columns
"""
return ObjectStream(resultTTree(self._ast, columns))
def visit_resultPandasDF(self, node):
'''
The result of this guy is an pandas dataframe. We generate a token here, and invoke the resultTTree in order to get the
actual ROOT file written. Later on, when dealing with the result stuff, we extract it into an awkward array.
'''
ttree = query_result_asts.resultTTree(node.source, node.column_names)
r = self.get_rep(ttree)
node.rep = rh.cpp_pandas_rep(r.filename, r.treename,
self._gc.current_scope())
self._result = node.rep
def test_TTree_columnNames_as_list():
r = resultTTree(None, ['hi', 'here'])
assert 2 == len(r.column_names)
def test_TTree_columnNames_as_string():
r = resultTTree(None, 'hi')
assert 1 == len(r.column_names)
def test_TTree_columnNames_as_tuple():
r = resultTTree(None, ('hi', 'here'))
assert 2 == len(r.column_names)