Ejemplo n.º 1
0
def test_inv_mix_columns_raises_exception_if_state_is_not_a_correct_array():
    with pytest.raises(ValueError):
        aes.inv_mix_columns(
            state=np.random.randint(0, 255, (12, 12), dtype='uint8'))
    with pytest.raises(ValueError):
        aes.inv_mix_columns(
            state=np.random.randint(0, 255, (12, 16), dtype='uint16'))
Ejemplo n.º 2
0
def test_inv_mix_columns_returns_correct_column_vector(aes_datas):
    state = aes_datas['input_state']
    expected = aes_datas['expected_inv_mix_columns']
    assert np.array_equal(
        expected,
        aes.inv_mix_columns(state)
    )
    assert expected.shape == state.shape
Ejemplo n.º 3
0
def test_inv_mix_columns_raises_exception_if_state_is_not_array():
    with pytest.raises(TypeError):
        aes.inv_mix_columns(state='foo')
    with pytest.raises(TypeError):
        aes.inv_mix_columns(state=12)