Commit 8c26c2a5 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add cross test

parent 7c798d47
Loading
Loading
Loading
Loading
+23 −1
Original line number Diff line number Diff line
import pytest
from hbutils.testing import tmatrix

from imgutils.data import is_valid_image_blob_url
from imgutils.data import is_valid_image_blob_url, load_image, to_blob_url, load_image_from_blob_url
from test.testings import get_testfile


@pytest.mark.unittest
class TestDataBlob:
    @pytest.mark.parametrize(*tmatrix({
        'filename': [
            'mostima_post.jpg',
            'soldiers.jpg',
            'nian.png',
        ],
        'format': [
            'jpg',
            'png',
            'webp',
        ]
    }, mode='matrix'))
    def test_to_blob_url_and_load_image_from_blob_url(self, filename, format, image_diff):
        original_image = load_image(get_testfile(filename), mode='RGB', force_background='white')
        assert image_diff(
            original_image,
            load_image_from_blob_url(to_blob_url(original_image, format=format)),
            throw_exception=False,
        ) < 1.5e-2

    @pytest.mark.parametrize(['blob_url', 'expected_result'], [
        ("data:image/png;base64,ABC", True),
        ("DATA:IMAGE/JPEG,", True),