Add tests for TH2 and TH3 arithmetic operations

Author: siliataider

bindings/pyroot/pythonizations/test/CMakeLists.txt

@@ -55,6 +55,7 @@ ROOT_ADD_PYUNITTEST(pyroot_pyz_ttree_branch ttree_branch.py PYTHON_DEPS numpy)
 ROOT_ADD_PYUNITTEST(pyroot_pyz_th1_operators th1_operators.py)
 ROOT_ADD_PYUNITTEST(pyroot_pyz_th1_fillN th1_fillN.py PYTHON_DEPS numpy)
 ROOT_ADD_PYUNITTEST(pyroot_pyz_th2 th2.py)
+ROOT_ADD_PYUNITTEST(pyroot_pyz_th3 th3.py)
 
 # TGraph, TGraph2D and error subclasses pythonizations
 ROOT_ADD_PYUNITTEST(pyroot_pyz_tgraph_getters tgraph_getters.py)

bindings/pyroot/pythonizations/test/th2.py

@@ -17,5 +17,42 @@ def test_GetBinError(self):
         self.assertEqual(h.GetBinErrorLow(1, 1), 2)
 
 
-if __name__ == '__main__':
+class TH2Operations(unittest.TestCase):
+    """
+    Test TH2D arithmetic operations
+    """
+
+    def setUp(self):
+        self.h1 = ROOT.TH2D("h1", "h1", 2, 0, 2, 2, 0, 2)
+        self.h2 = ROOT.TH2D("h2", "h2", 2, 0, 2, 2, 0, 2)
+
+        self.h1.Fill(0.5, 0.5, 2.0)
+        self.h2.Fill(0.5, 0.5, 3.0)
+
+    def test_addition(self):
+        hsum = self.h1 + self.h2
+        self.assertAlmostEqual(hsum.GetBinContent(1, 1), 5.0)
+
+    def test_subtraction(self):
+        hdiff = self.h2 - self.h1
+        self.assertAlmostEqual(hdiff.GetBinContent(1, 1), 1.0)
+
+    def test_multiplication(self):
+        hprod = self.h1 * self.h2
+        self.assertAlmostEqual(hprod.GetBinContent(1, 1), 6.0)
+
+    def test_division(self):
+        hdiv = self.h2 / self.h1
+        self.assertAlmostEqual(hdiv.GetBinContent(1, 1), 1.5)
+
+    def test_scalar_multiplication_left(self):
+        hscaled = 2.0 * self.h1
+        self.assertAlmostEqual(hscaled.GetBinContent(1, 1), 4.0)
+
+    def test_scalar_multiplication_right(self):
+        hscaled = self.h1 * 2.0
+        self.assertAlmostEqual(hscaled.GetBinContent(1, 1), 4.0)
+
+
+if __name__ == "__main__":
     unittest.main()

bindings/pyroot/pythonizations/test/th3.py

@@ -0,0 +1,44 @@
+import unittest
+
+import ROOT
+
+
+class TH3Operations(unittest.TestCase):
+    """
+    Test TH3D arithmetic operations
+    """
+
+    def setUp(self):
+        self.h1 = ROOT.TH3D("h1", "h1", 2, 0, 2, 2, 0, 2, 2, 0, 2)
+        self.h2 = ROOT.TH3D("h2", "h2", 2, 0, 2, 2, 0, 2, 2, 0, 2)
+
+        self.h1.Fill(0.5, 0.5, 0.5, 2.0)
+        self.h2.Fill(0.5, 0.5, 0.5, 3.0)
+
+    def test_addition(self):
+        hsum = self.h1 + self.h2
+        self.assertAlmostEqual(hsum.GetBinContent(1, 1, 1), 5.0)
+
+    def test_subtraction(self):
+        hdiff = self.h2 - self.h1
+        self.assertAlmostEqual(hdiff.GetBinContent(1, 1, 1), 1.0)
+
+    def test_multiplication(self):
+        hprod = self.h1 * self.h2
+        self.assertAlmostEqual(hprod.GetBinContent(1, 1, 1), 6.0)
+
+    def test_division(self):
+        hdiv = self.h2 / self.h1
+        self.assertAlmostEqual(hdiv.GetBinContent(1, 1, 1), 1.5)
+
+    def test_scalar_multiplication_left(self):
+        hscaled = 2.0 * self.h1
+        self.assertAlmostEqual(hscaled.GetBinContent(1, 1, 1), 4.0)
+
+    def test_scalar_multiplication_right(self):
+        hscaled = self.h1 * 2.0
+        self.assertAlmostEqual(hscaled.GetBinContent(1, 1, 1), 4.0)
+
+
+if __name__ == "__main__":
+    unittest.main()