summaryrefslogtreecommitdiff
path: root/src/text_recognizer/tests/test_line_predictor.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/text_recognizer/tests/test_line_predictor.py')
-rw-r--r--src/text_recognizer/tests/test_line_predictor.py35
1 files changed, 0 insertions, 35 deletions
diff --git a/src/text_recognizer/tests/test_line_predictor.py b/src/text_recognizer/tests/test_line_predictor.py
deleted file mode 100644
index eede4d4..0000000
--- a/src/text_recognizer/tests/test_line_predictor.py
+++ /dev/null
@@ -1,35 +0,0 @@
-"""Tests for LinePredictor."""
-import os
-from pathlib import Path
-import unittest
-
-
-import editdistance
-import numpy as np
-
-from text_recognizer.datasets import IamLinesDataset
-from text_recognizer.line_predictor import LinePredictor
-import text_recognizer.util as util
-
-SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "support"
-
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
-
-
-class TestEmnistLinePredictor(unittest.TestCase):
- """Test LinePredictor class on the EmnistLines dataset."""
-
- def test_filename(self) -> None:
- """Test that LinePredictor correctly predicts on single images, for several test images."""
- predictor = LinePredictor(
- dataset="EmnistLineDataset", network_fn="CNNTransformer"
- )
-
- for filename in (SUPPORT_DIRNAME / "emnist_lines").glob("*.png"):
- pred, conf = predictor.predict(str(filename))
- true = str(filename.stem)
- edit_distance = editdistance.eval(pred, true) / len(pred)
- print(
- f'Pred: "{pred}" | Confidence: {conf} | True: {true} | Edit distance: {edit_distance}'
- )
- self.assertLess(edit_distance, 0.2)