test_nst_image

Complexity

Total Complexity

6

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes

0

4 Functions

Rating	Name	Duplication	Size	Complexity
A	image_manager()	0	4	2
A	incompatible_image()	0	3	1
A	test_image_manager()	0	14	2
A	compatible_images()	0	5	1

import pytest


@pytest.fixture
def image_manager():
    from artificial_artwork.nst_image import ImageManager
    return ImageManager([lambda array: array + 2])


@pytest.fixture
def compatible_images(test_image):
    return type('CompatibleImages', (), {
        'content': test_image('canoe_water.jpg'),
        'style': test_image('blue-red-w400-h300.jpg'),
    })()

@pytest.fixture
def incompatible_image(test_image):
    return test_image('wikipedia-logo.png')


def test_image_manager(image_manager, compatible_images, incompatible_image):
    assert image_manager.images_compatible == False

    image_manager.load_from_disk(compatible_images.content, 'content')
    assert image_manager.images_compatible == False

    image_manager.load_from_disk(compatible_images.style, 'style')
    assert image_manager.images_compatible == True

    image_manager.load_from_disk(incompatible_image, 'content')
    assert image_manager.images_compatible == False

    with pytest.raises(ValueError):
        image_manager.load_from_disk(compatible_images.content, 'unknown-type')