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

Author: srittau

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
 ===================
 

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)

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):