imaginAIry/tests/test_safety.py

from PIL import Image

from imaginairy.safety import is_nsfw
from tests import TESTS_FOLDER


def test_is_nsfw():
    img = Image.open(f"{TESTS_FOLDER}/data/safety.jpg")
    assert is_nsfw(img)

    img = Image.open(f"{TESTS_FOLDER}/data/girl_with_a_pearl_earring.jpg")
    assert not is_nsfw(img)
tests: add more tests 2022-09-17 22:49:38 +00:00			`from PIL import Image`

			`from imaginairy.safety import is_nsfw`
			`from tests import TESTS_FOLDER`


			`def test_is_nsfw():`
			`img = Image.open(f"{TESTS_FOLDER}/data/safety.jpg")`
refactor: implements changes to comply with pylama 2022-09-22 17:56:18 +00:00			`assert is_nsfw(img)`
tests: add more tests 2022-09-17 22:49:38 +00:00
			`img = Image.open(f"{TESTS_FOLDER}/data/girl_with_a_pearl_earring.jpg")`
refactor: implements changes to comply with pylama 2022-09-22 17:56:18 +00:00			`assert not is_nsfw(img)`