You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
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)
|