def test_incorrect_columns(self):
"""Test if exception is raised in case there is 1
or no column.
"""
arr = [[0 for j in range(1)] for i in range(10)]
with pytest.raises(ValueError):
transfer.TransferFunction(arr)
def setup_class(self):
arr = [[1 for j in range(5)] for i in range(10)]
self.transfer = transfer.TransferFunction(arr)