diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2021-01-22 14:03:11 +0100 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2021-01-25 10:24:48 +0100 |
commit | 8af061ade39c07cbb26f921c42217a7bfdd1b6ba (patch) | |
tree | e887701f5396af179b40b678669d660cb59f6233 /ethosu/tensor_allocator/test/test_tensor_allocator.py | |
parent | ece4e65a786d61934d0c356169a3c5f11f188538 (diff) | |
download | ethos-u-vela-8af061ade39c07cbb26f921c42217a7bfdd1b6ba.tar.gz |
MLBEDSW-3832: Search allocator: improve C API
- Removed unnecessary casts
- Added more error handling
Change-Id: Ic822877544f67452339a20dca4addddc050d195c
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/tensor_allocator/test/test_tensor_allocator.py')
-rw-r--r-- | ethosu/tensor_allocator/test/test_tensor_allocator.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/ethosu/tensor_allocator/test/test_tensor_allocator.py b/ethosu/tensor_allocator/test/test_tensor_allocator.py index 5350fde0..1011279c 100644 --- a/ethosu/tensor_allocator/test/test_tensor_allocator.py +++ b/ethosu/tensor_allocator/test/test_tensor_allocator.py @@ -47,3 +47,17 @@ def test_allocate(lrs, expected_size): res = tensor_allocator.allocate(input, 0) assert len(res) == len(lrs) assert max(addr + lr[2] for addr, lr in zip(res, lrs)) == expected_size + + +def test_allocate_empty_input(): + assert [] == tensor_allocator.allocate([], 0) + + +invalid_input_test_data = [None, 3, [1, 2, 16, 9, 15], [1, 5, None], [-1, 0, 16], [1, 2, 10000000000]] + + +@pytest.mark.parametrize("input", invalid_input_test_data) +def test_allocate_invalid_input(input): + """Tests the search allocator with invalid input data""" + with pytest.raises(Exception): + tensor_allocator.allocate(input, 0) |