def _create_theme_from_xml(self, theme_xml, image_path): """ Return a theme object using information parsed from XML :param theme_xml: The Theme data object. :param image_path: Where the theme image is stored :return: Theme data. """ theme = ThemeXML() theme.parse(theme_xml) theme.extend_image_filename(image_path) return theme
def test_extend_image_filename(self): """ Test the extend_image_filename method """ # GIVEN: A theme object theme = ThemeXML() theme.theme_name = 'MyBeautifulTheme ' theme.background_filename = ' video.mp4' theme.background_type = 'video' path = os.path.expanduser('~') # WHEN: ThemeXML.extend_image_filename is run theme.extend_image_filename(path) # THEN: The filename of the background should be correct expected_filename = os.path.join(path, 'MyBeautifulTheme', 'video.mp4') self.assertEqual(expected_filename, theme.background_filename) self.assertEqual('MyBeautifulTheme', theme.theme_name)