Spaces:

junaid-simamdigital
/

SimamAlphaXR

Sleeping

SimamAlphaXR / tests /test_alpha_core.py

Add alpha tests

2be9278 verified about 1 month ago

308 Bytes

import numpy as np
from PIL import Image
from alpha_core import matte_for_frame


def test_green_screen_becomes_transparent():
    a = matte_for_frame(Image.fromarray(np.array([[[0, 255, 0], [255, 0, 0]]], dtype=np.uint8)), "Green screen")
    assert a.shape == (1, 2)
    assert int(a[0, 0]) < int(a[0, 1])

	import numpy as np
	from PIL import Image
	from alpha_core import matte_for_frame


	def test_green_screen_becomes_transparent():
	a = matte_for_frame(Image.fromarray(np.array([[[0, 255, 0], [255, 0, 0]]], dtype=np.uint8)), "Green screen")
	assert a.shape == (1, 2)
	assert int(a[0, 0]) < int(a[0, 1])