test_logo.py 463 Bytes
Newer Older
1 2
# coding=utf-8

3
"""weiss logo tests"""
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25

from numpy import uint8 
from sksurgeryimage.utilities.weisslogo import WeissLogo

def test_logo():
    """
    Test that we can make a nice looking logo
    """
    logo = WeissLogo()

    image = logo.get_logo()

    assert image.shape == (331, 331, 3)
    assert image.dtype == uint8

    logo = WeissLogo(400)
    image = logo.get_noisy_logo()

    assert image.shape == (400, 400, 3)
    assert image.dtype == uint8