Test hello world example binary can run

PiperOrigin-RevId: 312768125
Change-Id: I69e07f5ad797ae963e1802083f0fb50867a21713
This commit is contained in:
Pete Warden 2020-05-21 17:02:18 -07:00 committed by TensorFlower Gardener
parent 27d373215c
commit 8d021e4030
2 changed files with 39 additions and 0 deletions

View File

@ -91,3 +91,9 @@ cc_binary(
"//tensorflow/lite/schema:schema_fbs",
],
)
sh_test(
name = "hello_world_binary_test",
srcs = ["hello_world_binary_test.sh"],
data = [":hello_world"],
)

View File

@ -0,0 +1,33 @@
#!/bin/bash
# Copyright 2020 The TensorFlow Authors. All Rights Reserved.
#
# 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
#
# http://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.
# ==============================================================================
#
# Bash unit tests for the example binary.
set -e
OUTPUT_LOG_FILE=${TEST_TMPDIR}/output_log.txt
# Needed for copybara compatibility.
SCRIPT_BASE_DIR=/org_"tensor"flow
${TEST_SRCDIR}${SCRIPT_BASE_DIR}/tensorflow/lite/micro/examples/hello_world/hello_world 2>&1 | head > ${OUTPUT_LOG_FILE}
if ! grep -q 'x_value:.*y_value:' ${OUTPUT_LOG_FILE}; then
echo "ERROR: Expected logs not found in output '${OUTPUT_LOG_FILE}'"
exit 1
fi
echo
echo "SUCCESS: hello_world_binary_test PASSED"