def background_type(self, value): if isinstance(value, str): self.background_combo_box.setCurrentIndex( BackgroundType.from_string(value)) elif isinstance(value, int): self.background_combo_box.setCurrentIndex(value) else: raise TypeError( 'background_type must either be a string or an int')
def test_background_type_from_string(self): """ Test the from_string method of :class:`BackgroundType` """ # GIVEN: The BackgroundType strings background_type_solid = 'solid' background_type_gradient = 'gradient' background_type_image = 'image' background_type_transparent = 'transparent' background_type_video = 'video' background_type_stream = 'stream' # WHEN: Calling BackgroundType.from_string # THEN: The enum equivalents should be returned assert BackgroundType.from_string( background_type_solid) == BackgroundType.Solid assert BackgroundType.from_string( background_type_gradient) == BackgroundType.Gradient assert BackgroundType.from_string( background_type_image) == BackgroundType.Image assert BackgroundType.from_string( background_type_transparent) == BackgroundType.Transparent assert BackgroundType.from_string( background_type_video) == BackgroundType.Video assert BackgroundType.from_string( background_type_stream) == BackgroundType.Stream