run test with "pytest -s tests/tflm"

uTensor · Nov 29, 2019 · c034f7d · c034f7d
1 parent 5baed55
commit c034f7d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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'