Add tests for prediction code

Closes #126 (closed)

dan/predict/attention.py

@@ -70,14 +70,18 @@ def split_text_and_confidences(
         texts = list(text)
         offset = 0
     elif level == "word":
-        texts, probs = compute_prob_by_separator(text, confidences, word_separators)
+        texts, confidences = compute_prob_by_separator(
+            text, confidences, word_separators
+        )
         offset = 1
     elif level == "line":
-        texts, probs = compute_prob_by_separator(text, confidences, line_separators)
+        texts, confidences = compute_prob_by_separator(
+            text, confidences, line_separators
+        )
         offset = 1
     else:
         logger.error("Level should be either 'char', 'word', or 'line'")
-    return texts, [np.around(num, 2) for num in probs], offset
+    return texts, [np.around(num, 2) for num in confidences], offset
 
 
 def get_predicted_polygons_with_confidence(