You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
531 B
21 lines
531 B
import cv2
|
|
import service.colorSpaceService as colorSpaceService
|
|
|
|
img = cv2.imread('./img.jpg')
|
|
|
|
|
|
def test_getRGB():
|
|
r, g, b = colorSpaceService.getRGB([img])
|
|
_, _, _r = cv2.split(r)
|
|
_, _g, _ = cv2.split(g)
|
|
_b, _, _ = cv2.split(b)
|
|
assert (img == cv2.merge([_b, _g, _r])).all()
|
|
|
|
|
|
def test_getHSV():
|
|
h, s, v = colorSpaceService.getHSV([img])
|
|
_h, _, _ = cv2.split(h)
|
|
_, _s, _ = cv2.split(s)
|
|
_, _, _v = cv2.split(v)
|
|
assert (cv2.cvtColor(img, cv2.COLOR_BGR2HSV) == cv2.merge([_h, _s, _v])).all()
|