def get_title(self, url):
"""Get title for video from url given to url_info. Original title will
be fixed, and neccessary descriptions will be removed.
:return: Refactored title
"""
url_info = self.get_url_info(url)
return create_filename(url_info.get("title"))
def test_create_filename__small_letters(self):
result = create_filename("foo - bar")
self.assertEqual("Foo - Bar", result)
def test_create_filename__tags_to_remove(self):
result = create_filename("Foo - Bar[HD]")
self.assertEqual("Foo - Bar", result)
def test_create_filename__good_output(self):
result = create_filename("Foo - Bar")
self.assertEqual("Foo - Bar", result)