def test_pad_element_string(self):
with self.assertRaises(ValueError):
csvtable.pad_element("", 4, 0)
self.assertEqual("0000", csvtable.pad_element("", 4, "0"))
self.assertEqual("1200", csvtable.pad_element("12", 4, "0"))
self.assertEqual("1234", csvtable.pad_element("1234", 4, "0"))
self.assertEqual("12", csvtable.pad_element("1234", 2, "0"))
def test_pad_element_list(self):
self.assertListEqual([0, 0, 0, 0], csvtable.pad_element([], 4, 0))
self.assertListEqual([1, 2, 0, 0], csvtable.pad_element([1, 2], 4, 0))
self.assertListEqual([1, 2, 3, 4], csvtable.pad_element([1, 2, 3, 4], 4, 0))
self.assertListEqual([1, 2], csvtable.pad_element([1, 2, 3, 4], 2, 0))
