Python mix_columnの例

コード例 #1

ファイル: test_aes.py プロジェクト: ouzdeville/scared

def test_mix_column_raises_exception_if_state_is_not_a_correct_array():
    with pytest.raises(ValueError):
        aes.mix_column(
            vectors=np.random.randint(0, 255, (12, 12), dtype='uint8'))
    with pytest.raises(ValueError):
        aes.mix_column(
            vectors=np.random.randint(0, 255, (12, 4), dtype='uint16'))

コード例 #2

ファイル: test_aes.py プロジェクト: zongyuewang/scared

def test_mix_column_returns_correct_column_vector(aes_datas):
    state = aes_datas['input_vectors']
    expected = aes_datas['expected_mix_column']
    assert np.array_equal(
        expected,
        aes.mix_column(state)
    )
    assert expected.shape == state.shape

コード例 #3

ファイル: test_aes.py プロジェクト: ouzdeville/scared

def test_mix_column_raises_exception_if_state_is_not_array():
    with pytest.raises(TypeError):
        aes.mix_column(vectors='foo')
    with pytest.raises(TypeError):
        aes.mix_column(vectors=12)