Merge pull request #80 from Navneet78/func/join

Func/join
Hrabal · Oct 8, 2019 · 45916fe · 45916fe
2 parents 90f57f0 + d9236be
commit 45916fe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/tempy/tempy.py b/tempy/tempy.py
@@ -702,6 +702,13 @@ def clone(self):
         """Returns a deepcopy of this element."""
         return copy(self)
 
+    @classmethod
+    def join(cls, list):
+        n = len(list)
+        for index in range(1, 2*n-2, 2):
+            list.insert(index, cls())
+        return list
+
 
 class Escaped(DOMElement):
     def __init__(self, content, **kwargs):

diff --git a/tests/test_DOMElement.py b/tests/test_DOMElement.py
@@ -576,3 +576,9 @@ def test_find(self):
 
         result = tag.find('Br', names='foo')  # will return foo -> (Br())
         self.assertEqual(len(result), 1)
+
+    def test_join(self):
+        tag= Div()
+        list=['foo', 'Br', 'Div', 'Pre']
+        result= tag.join(list)
+        self.assertEqual(len(result), 7)