def test_inv_shift_rows_raises_exception_if_state_is_not_a_correct_array():
with pytest.raises(ValueError):
aes.inv_shift_rows(
state=np.random.randint(0, 255, (12, 12), dtype='uint8'))
with pytest.raises(ValueError):
aes.inv_shift_rows(
state=np.random.randint(0, 255, (12, 16), dtype='uint16'))
def test_inv_shift_rows_returns_correct_array(aes_datas):
state = aes_datas['input_state']
expected = aes_datas['expected_inv_shift_rows']
assert np.array_equal(
expected,
aes.inv_shift_rows(state=state)
)
assert expected.shape == state.shape
def test_inv_shift_rows_raises_exception_if_state_is_not_array():
with pytest.raises(TypeError):
aes.inv_shift_rows(state='foo')
with pytest.raises(TypeError):
aes.inv_shift_rows(state=12)