2 files changed, 5 insertions, 21 deletions
diff --git a/driver_library/python/test/test_driver.py b/driver_library/python/test/test_driver.py
index 28d0a29..0dd207f 100644
--- a/driver_library/python/test/test_driver.py
+++ b/driver_library/python/test/test_driver.py
@@ -67,12 +67,6 @@ def test_check_buffer_swig_ownership(network_buffer):
 
 @pytest.mark.parametrize('device_name', ['ethosu0'])
 @pytest.mark.parametrize('model_name', ['model.tflite'])
-def test_check_buffer_capacity(network_buffer):
-    assert network_buffer.capacity() > 0
-
-
-@pytest.mark.parametrize('device_name', ['ethosu0'])
-@pytest.mark.parametrize('model_name', ['model.tflite'])
 def test_check_buffer_size(network_buffer):
     assert network_buffer.size() > 0
 
@@ -81,16 +75,8 @@ def test_check_buffer_size(network_buffer):
 @pytest.mark.parametrize('model_name', ['model.tflite'])
 def test_check_buffer_clear(network_buffer):
     network_buffer.clear()
-    assert network_buffer.size() == 0
-
-
-@pytest.mark.parametrize('device_name', ['ethosu0'])
-@pytest.mark.parametrize('model_name', ['model.tflite'])
-def test_check_buffer_resize(network_buffer):
-    offset = 1
-    new_size = network_buffer.capacity() - offset
-    network_buffer.resize(new_size, offset)
-    assert network_buffer.size() == new_size
+    for i in range(network_buffer.size()):
+        assert network_buffer.data()[i] == 0
 
 
 @pytest.mark.parametrize('device_name', ['ethosu0'])
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'])