diff options
Diffstat (limited to 'driver_library/python/test/test_driver.py')
-rw-r--r-- | driver_library/python/test/test_driver.py | 18 |
1 files changed, 2 insertions, 16 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']) |