def test_table_add_css_classes_to_column(self): t = Table(('Key', 'Value')) t.add_row(Key="Employee name", Value="John Doe") t.add_row(Key="D.O.B", Value="12/06/1982") self.assertEqual( t.html(), "<table>\n" "<tr><th>Key</th><th>Value</th></tr>\n" "<tr><td>Employee name</td><td>John Doe</td></tr>\n" "<tr><td>D.O.B</td><td>12/06/1982</td></tr>\n" "</table>") t.add_css_classes("key-column", column="Key") self.assertEqual( t.html(), "<table>\n" "<tr><th class='key-column'>Key</th><th>Value</th></tr>\n" "<tr><td class='key-column'>Employee name</td><td>John Doe</td></tr>\n" "<tr><td class='key-column'>D.O.B</td><td>12/06/1982</td></tr>\n" "</table>") t.add_css_classes("summary", "sunny", column="Key") self.assertEqual( t.html(), "<table>\n" "<tr><th class='key-column summary sunny'>Key</th><th>Value</th></tr>\n" "<tr><td class='key-column summary sunny'>Employee name</td><td>John Doe</td></tr>\n" "<tr><td class='key-column summary sunny'>D.O.B</td><td>12/06/1982</td></tr>\n" "</table>") self.assertRaises(KeyError, t.add_css_classes, "blah", column="doesntexist")
def test_table_add_css_classes(self): t = Table(('Key', 'Value')) t.add_row(Key="Employee name", Value="John Doe") self.assertEqual( t.html(), "<table>\n" "<tr><th>Key</th><th>Value</th></tr>\n" "<tr><td>Employee name</td><td>John Doe</td></tr>\n" "</table>") t.add_css_classes("employee-data") self.assertEqual( t.html(), "<table class='employee-data'>\n" "<tr><th>Key</th><th>Value</th></tr>\n" "<tr><td>Employee name</td><td>John Doe</td></tr>\n" "</table>") t.add_css_classes("summary", "sunny") self.assertEqual( t.html(), "<table class='employee-data summary sunny'>\n" "<tr><th>Key</th><th>Value</th></tr>\n" "<tr><td>Employee name</td><td>John Doe</td></tr>\n" "</table>")