def test__get_raw_image(self): Gallery._raw_images["existing_key"] = "data" self.assertEqual("data", Gallery._get_raw_image("existing_key"))
def test__get_raw_image_non_existing_key(self, pillow_open): self.assertEqual("data", Gallery._get_raw_image("non_existing_key")) pillow_open.assert_called_once_with("img/non_existing_key.png")