Jeremy Johnson | 48df8c7 | 2023-09-12 14:52:34 +0100 | [diff] [blame] | 1 | """Tests for the python interface to the verifier library.""" |
| 2 | # Copyright (c) 2023, ARM Limited. |
| 3 | # SPDX-License-Identifier: Apache-2.0 |
| 4 | from pathlib import Path |
| 5 | |
| 6 | import numpy as np |
| 7 | import pytest |
| 8 | from checker.verifier import VerifierError |
| 9 | from checker.verifier import VerifierLibrary |
| 10 | |
| 11 | # NOTE: These tests are set to POST COMMIT - so will only run on the CI |
| 12 | |
| 13 | # Location of reference model binaries |
| 14 | REF_MODEL_BUILD_PATH = Path(__file__).resolve().parents[2] / "build" / "reference_model" |
| 15 | VERIFIER_LIB = "libtosa_reference_verify_lib.so" |
| 16 | VERIFIER_LIB_PATH = REF_MODEL_BUILD_PATH / VERIFIER_LIB |
| 17 | |
| 18 | |
| 19 | @pytest.mark.postcommit |
| 20 | def test_verifier_lib_built(): |
| 21 | """First test to check the library has been built.""" |
| 22 | assert VERIFIER_LIB_PATH.is_file() |
| 23 | |
| 24 | |
| 25 | @pytest.mark.postcommit |
| 26 | def test_checker_verifier_load_fail(): |
| 27 | with pytest.raises(VerifierError) as excinfo: |
| 28 | VerifierLibrary(Path("/place-that-does-not-exist")) |
Jeremy Johnson | f0348ea | 2023-09-27 16:10:59 +0100 | [diff] [blame] | 29 | assert str(excinfo.value).startswith("Could not find verify library") |
Jeremy Johnson | 48df8c7 | 2023-09-12 14:52:34 +0100 | [diff] [blame] | 30 | |
| 31 | |
| 32 | @pytest.mark.postcommit |
| 33 | def test_checker_verifier_load(): |
Jeremy Johnson | f0348ea | 2023-09-27 16:10:59 +0100 | [diff] [blame] | 34 | vlib = VerifierLibrary(VERIFIER_LIB_PATH) |
Jeremy Johnson | 48df8c7 | 2023-09-12 14:52:34 +0100 | [diff] [blame] | 35 | assert vlib |
| 36 | |
| 37 | |
| 38 | JSON_COMPLIANCE_DOT_PRODUCT = { |
| 39 | "version": "0.1", |
| 40 | "tensors": { |
| 41 | "output1": { |
| 42 | "mode": "DOT_PRODUCT", |
Jeremy Johnson | bb0935f | 2023-09-14 16:43:48 +0100 | [diff] [blame] | 43 | "data_type": "FP32", |
| 44 | "dot_product_info": {"ks": 1000, "s": 0}, |
Jeremy Johnson | 48df8c7 | 2023-09-12 14:52:34 +0100 | [diff] [blame] | 45 | } |
| 46 | }, |
| 47 | } |
| 48 | |
| 49 | |
| 50 | @pytest.mark.postcommit |
| 51 | def test_checker_verifier_dot_product_check(): |
Jeremy Johnson | f0348ea | 2023-09-27 16:10:59 +0100 | [diff] [blame] | 52 | vlib = VerifierLibrary(VERIFIER_LIB_PATH) |
Jeremy Johnson | 48df8c7 | 2023-09-12 14:52:34 +0100 | [diff] [blame] | 53 | assert vlib |
| 54 | |
| 55 | imp_arr = np.zeros((10, 10, 10), dtype=np.float32) |
| 56 | ref_arr = np.zeros((10, 10, 10), dtype=np.float64) |
| 57 | bnd_arr = np.zeros((10, 10, 10), dtype=np.float64) |
| 58 | |
| 59 | json_config = JSON_COMPLIANCE_DOT_PRODUCT |
| 60 | |
| 61 | ret = vlib.verify_data("output1", json_config, imp_arr, ref_arr, bnd_arr) |
| 62 | assert ret |
| 63 | |
| 64 | |
| 65 | @pytest.mark.postcommit |
| 66 | def test_checker_verifier_dot_product_check_fail(): |
Jeremy Johnson | f0348ea | 2023-09-27 16:10:59 +0100 | [diff] [blame] | 67 | vlib = VerifierLibrary(VERIFIER_LIB_PATH) |
Jeremy Johnson | 48df8c7 | 2023-09-12 14:52:34 +0100 | [diff] [blame] | 68 | assert vlib |
| 69 | |
| 70 | imp_arr = np.zeros((10, 10, 10), dtype=np.float32) |
| 71 | ref_arr = np.ones((10, 10, 10), dtype=np.float64) |
| 72 | bnd_arr = np.zeros((10, 10, 10), dtype=np.float64) |
| 73 | |
| 74 | json_config = JSON_COMPLIANCE_DOT_PRODUCT |
| 75 | |
| 76 | ret = vlib.verify_data("output1", json_config, imp_arr, ref_arr, bnd_arr) |
| 77 | assert not ret |