From d0eee26bc17ecd237c1b1e86cda78f5f310af391 Mon Sep 17 00:00:00 2001 From: Diego Russo Date: Thu, 23 Apr 2020 18:14:37 +0100 Subject: Add test for len1_array_to_scalar function Moved len1_array_to_scalar from a nested function to a staticmethod of TFLiteSubgraph. Change-Id: I182f0b70f03070855c1a4478d26644892c1ebb15 Signed-off-by: Diego Russo --- ethosu/vela/test/test_tflite_reader.py | 36 ++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 ethosu/vela/test/test_tflite_reader.py (limited to 'ethosu/vela/test') diff --git a/ethosu/vela/test/test_tflite_reader.py b/ethosu/vela/test/test_tflite_reader.py new file mode 100644 index 00000000..898e3840 --- /dev/null +++ b/ethosu/vela/test/test_tflite_reader.py @@ -0,0 +1,36 @@ +# Copyright (C) 2020 Arm Limited or its affiliates. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the License); you may +# not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an AS IS BASIS, WITHOUT +# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# Description: +# Contains unit tests for tflite_reader +import pytest +from ethosu.vela.tflite_reader import TFLiteSubgraph + + +class TestTFLiteSubgraph: + + # Generate some data for testing len1_array_to_scalar + len1_testdata = [ + (0, None), + pytest.param(1, None, marks=pytest.mark.xfail), + ([1, 2, 3], [1, 2, 3]), + ([10], 10), + ([], []), + ] + + @pytest.mark.parametrize("test_input,expected", len1_testdata) + def test_len1_array_to_scalar(self, test_input, expected): + output = TFLiteSubgraph.len1_array_to_scalar(test_input) + assert output == expected -- cgit v1.2.1