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