Skip to content
Snippets Groups Projects
Commit f3a41a88 authored by Enric Tejedor Saavedra's avatar Enric Tejedor Saavedra
Browse files

[Exp PyROOT] Pass file open mode as parameter

parent d5933e0e
No related branches found
No related tags found
No related merge requests found
......@@ -11,9 +11,10 @@ struct MyStruct {
// - vectorb: branch of type `std::vector<double>`, size `arraysize`
// - structb: struct branch of type `MyStruct`
// - structleafb: struct branch of type `MyStruct`, created as a leaf list
void CreateTTree(const char *filename, const char *treename, int nentries, int arraysize, int more)
void CreateTTree(const char *filename, const char *treename, int nentries, int arraysize, int more,
const char* openmode)
{
TFile f(filename, "RECREATE");
TFile f(filename, openmode);
TTree t(treename, "Test tree");
// Float branch
......
......@@ -20,7 +20,12 @@ class TTreeBranchAttr(unittest.TestCase):
@classmethod
def setUpClass(cls):
ROOT.gInterpreter.Declare('#include "TreeHelper.h"')
ROOT.CreateTTree(cls.filename, cls.treename, cls.nentries, cls.arraysize, cls.more)
ROOT.CreateTTree(cls.filename,
cls.treename,
cls.nentries,
cls.arraysize,
cls.more,
"RECREATE")
# Helper
def get_file_and_tree(self):
......
......@@ -23,7 +23,12 @@ class TTreeIterable(unittest.TestCase):
@classmethod
def setUpClass(cls):
ROOT.gInterpreter.Declare('#include "TreeHelper.h"')
ROOT.CreateTTree(cls.filename, cls.treename, cls.nentries, cls.arraysize, cls.more)
ROOT.CreateTTree(cls.filename,
cls.treename,
cls.nentries,
cls.arraysize,
cls.more,
"RECREATE")
# Helper
def get_file_and_tree(self):
......
......@@ -27,7 +27,12 @@ class TTreeSetBranchAddress(unittest.TestCase):
@classmethod
def setUpClass(cls):
ROOT.gInterpreter.Declare('#include "TreeHelper.h"')
ROOT.CreateTTree(cls.filename, cls.treename, cls.nentries, cls.arraysize, cls.more)
ROOT.CreateTTree(cls.filename,
cls.treename,
cls.nentries,
cls.arraysize,
cls.more,
"RECREATE")
# Helper
def get_file_tree_and_chain(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment