[test] stl vector __iadd__ with array.array

Author: aaronj0

bindings/pyroot/pythonizations/test/stl_vector.py

@@ -45,6 +45,32 @@ def test_stl_vector_boolean(self):
             self.assertFalse(vector.empty())
             self.assertTrue(bool(vector))
 
+    def test_stl_vector_iadd(self):
+        import random
+        import array
+        """
+        Test that the __iadd__ pythonization of std::vector works as expected.
+        This call dispatches to std::insert
+        https://github.com/root-project/root/issues/
+        """
+        def assertVec(vec, arr):
+            for x, y in zip(vec, arr):
+                self.assertEqual(x, y)
+
+        entry_type = ['unsigned short', 'short', 'int', 'unsigned int', 'long', 'unsigned long', 'float', 'double']
+        array_type = ['H', 'h', 'i', 'I', 'l', 'L', 'f', 'd']
+
+        typemap = zip(entry_type, array_type)
+        n = 5
+        for dtype in typemap:
+            vector = ROOT.std.vector[dtype[0]]()
+            self.assertTrue(vector.empty())
+            li = [random.randint(1, 100) for _ in range(n)]
+            arr = array.array(dtype[1], li)
+            vec += arr
+            self.assertFalse(vector.empty())
+            assertVec(vec, arr)
+
     def test_tree_with_containers(self):
         """
         Test that the boolean conversion of a std::vector works as expected inside a TTree.