def test_add_line_data(self):
"""Does this work with correct input"""
input = "F3W.3.Dr.002 6/29/2013 450"
expected = "F3W.3.Dr.002 6/29/2013 4:50:00 no_data flagstaff 3 wall 3 drywall 002 1 no_data"
self.assertEqual(add_line_data(input), expected)
def test_add_line_data1(self):
"""Does this work with correct input"""
input = "T3C.3.Ce.016 8/2/2013 13:43:00 JS 3"
expected = "T3C.3.Ce.016 8/2/2013 13:43:00 JS toronto 3 ceiling 3 ceiling 016 3 3"
self.assertEqual(add_line_data(input), expected)
def test_add_line_data2(self):
"""Does this work with correct input"""
input = "F2F.3.Ca.009 "
expected = "F2F.3.Ca.009 no_data no_data no_data flagstaff 2 floor 3 carpet 009 2 no_data"
self.assertEqual(add_line_data(input), expected)
