def test_b64_string_to_ndarray():
from image_processor_client import image_file_to_b64
from image_processor_client import b64_string_to_ndarray
b64 = image_file_to_b64("images/acl1_test.jpg")
nd = b64_string_to_ndarray(b64)
answer = nd[25][0:5]
expected = [[5, 5, 5], [5, 5, 5], [5, 5, 5], [5, 5, 5], [5, 5, 5]]
assert (answer == expected).all
def test_ndarray_to_b64_string():
from image_processor_client import image_file_to_b64
from image_processor_server import b64_string_to_ndarray
from image_processor_server import ndarray_to_b64_string
b64 = image_file_to_b64("images/acl2_test.jpg")
nd = b64_string_to_ndarray(b64)
answer = ndarray_to_b64_string(nd)
expected = 'iVBORw0KGgoAAAANSUhE'
assert answer[0:20] == expected
def test_process_image_inversion():
from image_processor_client import image_file_to_b64
from image_processor_server import b64_string_to_ndarray
from image_processor_server import process_image_inversion
b64 = image_file_to_b64("images/acl1_test.jpg")
nd = b64_string_to_ndarray(b64)
inverted_nd = process_image_inversion(nd)
answer = inverted_nd[25][0:5]
expected = [[5, 5, 5], [5, 5, 5], [5, 5, 5], [5, 5, 5], [5, 5, 5]]
assert (answer == expected).all
def test_b64_to_image_file():
from image_processor_client import image_file_to_b64
from image_processor_client import b64_to_image_file
import filecmp
import os
b64str = image_file_to_b64("images/acl1_test.jpg")
b64_to_image_file(b64str, "images/test_output.jpg")
answer = filecmp.cmp("images/acl1_test.jpg", "images/test_output.jpg")
os.remove("images/test_output.jpg")
expected = True
assert answer == expected
def test_image_file_to_b64_wrongfiletype():
from image_processor_client import image_file_to_b64
answer = image_file_to_b64("images/test.txt")
expected = ""
assert answer == expected
def test_image_file_to_b64_notfound():
from image_processor_client import image_file_to_b64
answer = image_file_to_b64("images/acl.jpg")
expected = False
assert answer == expected
def test_image_file_to_b64_exists():
from image_processor_client import image_file_to_b64
answer = image_file_to_b64("images/acl1.jpg")
assert answer[0:20] == "/9j/4AAQSkZJRgABAgAA"
