def correctly_build_movie(lines):
title, duration = parse_first_line(lines[0])
movie = Movie(title, duration)
for line in lines[1:]:
movie = parse_movie_information(movie, line)
return movie
def test_different_timing(self):
line = " Big Fish 2h5m"
title, duration = parse_first_line(line)
self.assertEqual(title, "Big Fish")
self.assertEqual(duration, "2h5m")
def test_watched_movie(self):
line = " '''The Grand Budapest Hotel''' 1h39m"
title, duration = parse_first_line(line)
self.assertEqual(title, "The Grand Budapest Hotel")
self.assertEqual(duration, "1h39m")
def test_series(self):
line = " Kung Fu Panda (series) (2h56m)"
title, duration = parse_first_line(line)
self.assertEqual(title, "Kung Fu Panda series")
self.assertEqual(duration, "2h56m")
def test_different_indentation(self):
line = " The Godfather 1 2h55m"
title, duration = parse_first_line(line)
self.assertEqual(title, "The Godfather 1")
self.assertEqual(duration, "2h55m")
def test_normal_line(self):
line = " Whiplash 1h49m"
title, duration = parse_first_line(line)
self.assertEqual(title, "Whiplash")
self.assertEqual(duration, "1h49m")