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"