diff options
Diffstat (limited to 'driver_library/python/test/test_driver_utilities.py')
-rw-r--r-- | driver_library/python/test/test_driver_utilities.py | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/driver_library/python/test/test_driver_utilities.py b/driver_library/python/test/test_driver_utilities.py index fc8e921..fe44b0e 100644 --- a/driver_library/python/test/test_driver_utilities.py +++ b/driver_library/python/test/test_driver_utilities.py @@ -1,5 +1,5 @@ # -# SPDX-FileCopyrightText: Copyright 2021-2022 Arm Limited and/or its affiliates <open-source-office@arm.com> +# SPDX-FileCopyrightText: Copyright 2021-2023 Arm Limited and/or its affiliates <open-source-office@arm.com> # SPDX-License-Identifier: Apache-2.0 # import pytest @@ -53,10 +53,8 @@ def test_check_network_ifm_size(network): def test_allocate_buffers(device): buffers = driver.allocate_buffers(device, [128, 256]) assert len(buffers) == 2 - assert buffers[0].size() == 0 - assert buffers[0].capacity() == 128 - assert buffers[1].size() == 0 - assert buffers[1].capacity() == 256 + assert buffers[0].size() == 128 + assert buffers[1].size() == 256 @pytest.mark.parametrize('device_name', ['ethosu0']) |