[Exp PyROOT] Add tests for TDirectory pythonizations

bindings/pyroot_experimental/PyROOT/test/CMakeLists.txt

@@ -3,6 +3,7 @@ ROOT_ADD_PYUNITTEST(pyroot_pyz_pretty_printing pretty_printing.py)
 ROOT_ADD_PYUNITTEST(pyroot_pyz_array_interface array_interface.py)
 
 # TFile, TDirectory, TDirectoryFile pythonizations
+ROOT_ADD_PYUNITTEST(pyroot_pyz_tdirectory_read_write tdirectoryfile_read_write.py)
 ROOT_ADD_PYUNITTEST(pyroot_pyz_tdirectoryfile_read_write tdirectoryfile_read_write.py)
 ROOT_ADD_PYUNITTEST(pyroot_pyz_tfile_open_read_write tfile_open_read_write.py)
 

bindings/pyroot_experimental/PyROOT/test/tdirectory_read_write.py

+import unittest
+
+import ROOT
+from libcppyy import SetOwnership
+
+
+class TDirectoryReadWrite(unittest.TestCase):
+    """
+    Test for the attr syntax and Get method of TDirectory.
+    """
+
+    nbins = 8
+    xmin = 0
+    xmax = 4
+
+    # Setup
+    @classmethod
+    def setUpClass(cls):
+        cls.dir0 = ROOT.TDirectory("dir0", "dir0")
+        h = ROOT.TH1F("h", "h", cls.nbins, cls.xmin, cls.xmax)
+        SetOwnership(h, False)
+        # this must be there otherwise the histogram is not attached to dir0
+        h.SetDirectory(cls.dir0)
+
+        dir1 = cls.dir0.mkdir("dir1")
+        dir1.cd()
+        h1 = ROOT.TH1F("h1", "h1", cls.nbins, cls.xmin, cls.xmax)
+        SetOwnership(h1, False)
+
+        dir2 = dir1.mkdir("dir2")
+        dir2.cd()
+        h2 = ROOT.TH1F("h2", "h2", cls.nbins, cls.xmin, cls.xmax)
+        SetOwnership(h2, False)
+
+    def checkHisto(self, h):
+        xaxis = h.GetXaxis()
+        self.assertEqual(self.nbins, h.GetNbinsX())
+        self.assertEqual(self.xmin, xaxis.GetXmin())
+        self.assertEqual(self.xmax, xaxis.GetXmax())
+
+    # Tests
+    def test_readHisto_attrsyntax(self):
+        self.checkHisto(self.dir0.h)
+        self.checkHisto(self.dir0.dir1.h1)
+        self.checkHisto(self.dir0.dir1.dir2.h2)
+
+if __name__ == '__main__':
+    unittest.main()