def test_short(self):
in_string = "1234567"
out_string = improve_deptid(in_string)
self.assertEqual("01234567", out_string)
def test_invalid_data(self):
in_string = "A6"
with self.assertRaises(InvalidDataException):
out_string = improve_deptid(in_string)
print out_string
def test_no_op(self):
in_string = "16350100"
out_string = improve_deptid(in_string)
self.assertEqual(in_string, out_string)