def test_image_too_large():
"""Error if the image is larger than the set width"""
with pytest.raises(InvalidCombinationOfParameters):
generate_numbers_sequence([1, 2, 3, 4, 5, 6, 7, 7], (1, 3), 50)
def test_max_range(self, spacing_range):
"""Spaces larger than the image."""
with pytest.raises(MaxSpaceTooLarge):
generate_numbers_sequence(digits, spacing_range, image_width)
import numpy as np
import pytest
from script import generate_numbers_sequence
from script import InvalidCombinationOfParameters
image_width, spacing_range, digits = 100, (1, 3), [1, 2]
image = generate_numbers_sequence(digits, spacing_range, image_width)
def test_output_type():
assert isinstance(image, np.ndarray)
def test_size():
height = image.shape[0] == 28
width = image.shape[1] == image_width
assert all([height, width])
def test_image_too_large():
"""Error if the image is larger than the set width"""
with pytest.raises(InvalidCombinationOfParameters):
generate_numbers_sequence([1, 2, 3, 4, 5, 6, 7, 7], (1, 3), 50)
def test_range(self, spacing_range):
"""Error when max range not larger than min range"""
with pytest.raises(InvalidSpacingRange):
generate_numbers_sequence(digits, spacing_range, image_width)
def test_negative_range(self, spacing_range):
"""Error when min range negative"""
with pytest.raises(InvalidSpacingValue):
generate_numbers_sequence(digits, spacing_range, image_width)
def test_each_element_type(self, spacing_range):
"""Error if at least one element is not integer"""
with pytest.raises(ValuesNotInteger):
generate_numbers_sequence(digits, spacing_range, image_width)
def test_input_size(self, spacing_range):
"""Error if input size not of correct size, has to be size 2"""
with pytest.raises(SpacingInvalidSize):
generate_numbers_sequence(digits, spacing_range, image_width)
def test_input_type(self, spacing_range):
"""Error if wrong type, has to be tuple"""
with pytest.raises(SpacingNotTuple):
generate_numbers_sequence(digits, spacing_range, image_width)
def test_invalid_value(self, image_width):
"""Error if negative or zero"""
with pytest.raises(InvalidImageWidth):
generate_numbers_sequence(digits, spacing_range, image_width)
def test_input_type(self, image_width):
"""Error if not integer"""
with pytest.raises(ImageWidthNotInteger):
generate_numbers_sequence(digits, spacing_range, image_width)
