fixed error when passing elements to TableCell's and TableHeaderCell'…

…s constructor
srittau · Oct 22, 2014 · e17c46c · e17c46c
1 parent b030558
commit e17c46c
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/NEWS.rst b/NEWS.rst
@@ -1,3 +1,12 @@
+News in version 0.6.1
+=====================
+
+Bug Fixes
+---------
+
+  * Fixed error when passing elements to TableCell's and TableHeaderCell's
+    constructor.
+
 News in version 0.6
 ===================
 

diff --git a/htmlgen/table.py b/htmlgen/table.py
@@ -275,7 +275,7 @@ class _TableCellBase(Element):
 
     def __init__(self, element_name, *content):
         super(_TableCellBase, self).__init__(element_name)
-        self.extend(*content)
+        self.extend(content)
 
     rows = int_html_attribute("rowspan", 1)
     columns = int_html_attribute("colspan", 1)

diff --git a/test_htmlgen/table.py b/test_htmlgen/table.py
@@ -2,7 +2,7 @@
 
 from asserts import assert_equal, assert_true
 
-from htmlgen import Table, TableHead, TableRow, TableCell, ColumnGroup
+from htmlgen import Table, TableHead, TableRow, TableCell, ColumnGroup, Span
 
 
 class TableTest(TestCase):
@@ -154,6 +154,10 @@ def test_columns_and_rows(self):
         cell.rows = 5
         assert_equal('<td colspan="3" rowspan="5">Content</td>', str(cell))
 
+    def test_element_child(self):
+        cell = TableCell(Span("Content"))
+        assert_equal('<td><span>Content</span></td>', str(cell))
+
 
 class ColumnGroupTest(TestCase):