diff options
author | Cathal Corbett <cathal.corbett@arm.com> | 2021-11-23 11:49:23 +0000 |
---|---|---|
committer | Cathal Corbett <cathal.corbett@arm.com> | 2021-11-23 16:52:46 +0000 |
commit | bddb2583f2fe3529787f55a4143b24db42951721 (patch) | |
tree | 37061e1bb32cc1d466cc04cdfbd85665c8542aa0 /python | |
parent | f7b5011298367c6d635b17ad029c627076072198 (diff) | |
download | armnn-bddb2583f2fe3529787f55a4143b24db42951721.tar.gz |
IVGCVSW-6610 No bounds checking performed when indexing PermutationVector elements.
* Added out of bounds index error checking to PermutationVector operator[] method
in Types.hpp.
* Added armnn unit tests to UnitsTests.cpp to check error when using
PermutationVector alone and with TransposeDescriptor and PermuteDescriptor.
* Added pyarmnn unit test to check error when using PermutationVector.
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ie5a8090f07b571a6bdf79a8cad621c31cc1891b7
Diffstat (limited to 'python')
-rw-r--r-- | python/pyarmnn/test/test_types.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/python/pyarmnn/test/test_types.py b/python/pyarmnn/test/test_types.py index dfe1429c02..500a779844 100644 --- a/python/pyarmnn/test/test_types.py +++ b/python/pyarmnn/test/test_types.py @@ -27,3 +27,8 @@ def test_permutation_vector(): pv4 = ann.PermutationVector((0, 3, 1, 2)) assert pv.IsInverse(pv4) + + with pytest.raises(ValueError) as err: + pv4[4] + + assert err.type is ValueError |