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'))
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
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)