| 12345678910111213141516171819 |
- from sorawm.iopaint.helper import load_img
- from sorawm.iopaint.tests.utils import current_dir
- png_img_p = current_dir / "image.png"
- jpg_img_p = current_dir / "bunny.jpeg"
- def test_load_png_image():
- with open(png_img_p, "rb") as f:
- np_img, alpha_channel = load_img(f.read())
- assert np_img.shape == (256, 256, 3)
- assert alpha_channel.shape == (256, 256)
- def test_load_jpg_image():
- with open(jpg_img_p, "rb") as f:
- np_img, alpha_channel = load_img(f.read())
- assert np_img.shape == (394, 448, 3)
- assert alpha_channel is None
|
