bug fix

ClarityNLP · Aug 5, 2019 · c21cf58 · c21cf58
1 parent 0a4e0cf
commit c21cf58
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/nlp/algorithms/finder/test_finder.py b/nlp/algorithms/finder/test_finder.py
@@ -153,7 +153,7 @@ def _run_tests(module_type, test_data):
             computed_values = [tf.TimeValue(**d) for d in json_data]
 
             # check computed vs. expected results
-            return _compare_results(
+            ok =_compare_results(
                 computed_values,
                 expected_values,
                 sentence,
@@ -166,7 +166,7 @@ def _run_tests(module_type, test_data):
             json_data = json.loads(json_result)
             computed_values = [df.DateValue(**d) for d in json_data]
 
-            return _compare_results(
+            ok = _compare_results(
                 computed_values,
                 expected_values,
                 sentence,
@@ -179,12 +179,17 @@ def _run_tests(module_type, test_data):
             json_data = json.loads(json_result)
             computed_values = [smf.SizeMeasurement(**d) for d in json_data]
 
-            return _compare_results(
+            ok =_compare_results(
                 computed_values,
                 expected_values,
                 sentence,
                 _SIZE_MEAS_FIELDS)
 
+        if not ok:
+            return False
+
+    return True
+
 
 ###############################################################################
 def test_time_finder():