fix tests for rx.image

reflex-dev · Oct 4, 2023 · 9a63e1b · 9a63e1b
1 parent 6377d66
commit 9a63e1b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/components/media/test_image.py b/tests/components/media/test_image.py
@@ -34,7 +34,7 @@ def test_serialize_image(pil_image: Img):
     def test_set_src_str():
         """Test that setting the src works."""
         image = rx.image(src="pic2.jpeg")
-        assert str(image.src) == "pic2.jpeg"  # type: ignore
+        assert str(image.src) == "{`pic2.jpeg`}"  # type: ignore
 
     def test_set_src_img(pil_image: Img):
         """Test that setting the src works.
@@ -43,7 +43,7 @@ def test_set_src_img(pil_image: Img):
             pil_image: The image to serialize.
         """
         image = Image.create(src=pil_image)
-        assert str(image.src) == serialize_image(pil_image)  # type: ignore
+        assert str(image.src.name) == serialize_image(pil_image)  # type: ignore
 
     def test_render(pil_image: Img):
         """Test that rendering an image works.
@@ -52,8 +52,6 @@ def test_render(pil_image: Img):
             pil_image: The image to serialize.
         """
         image = Image.create(src=pil_image)
-        assert not image.src.is_string  # type: ignore
-        image._render()
         assert image.src.is_string  # type: ignore
 
 except ImportError: