From c034f7de86d3ebf33bb39ccc59a550b03eb9e77e Mon Sep 17 00:00:00 2001 From: Neil Tan Date: Thu, 28 Nov 2019 22:24:04 -0800 Subject: [PATCH] run test with "pytest -s tests/tflm" --- tests/tflm/tflite_export/test_write.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/tflm/tflite_export/test_write.py b/tests/tflm/tflite_export/test_write.py index b905fb77..c5930e71 100644 --- a/tests/tflm/tflite_export/test_write.py +++ b/tests/tflm/tflite_export/test_write.py @@ -110,5 +110,6 @@ def test_tflite_fb_write(hybrid_quant_output): print("out2 :", test_model.get_tensor(test_model.get_output_details()[2]["index"])) - test_pass = True - assert test_pass, 'error message here' + output = test_model.get_tensor(test_model.get_output_details()[2]["index"]) + + assert np.abs(output - 707) <= 0.0001, 'error is greater than 0.0001'