Commit 9e48ab33 authored by Stephen Thompson's avatar Stephen Thompson

Issue #6 removed logo tests

parent 00957dae
Pipeline #3272 passed with stages
in 7 minutes and 7 seconds
# coding=utf-8
"""snappysonic tests"""
from numpy import uint8, zeros
from snappysonic.algorithms.logo import WeissLogo, noisy
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
def test_noisy():
"""
tests function to add noise to a colour image.
"""
timage = zeros((10, 10, 1), uint8)
_ = noisy(timage)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment