summaryrefslogtreecommitdiff
path: root/tests/test_character_predictor.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-02-06 22:42:42 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-02-06 22:42:42 +0100
commit2d4e527fc0d47db8ed995180cbcec01271e20245 (patch)
treeb3d8266191632df5c0668bb6eb56bc5cdc2def90 /tests/test_character_predictor.py
parent1e09d9d340fd214ee48fb093d51c67a415b3be19 (diff)
chore: remove tests
Diffstat (limited to 'tests/test_character_predictor.py')
-rw-r--r--tests/test_character_predictor.py31
1 files changed, 0 insertions, 31 deletions
diff --git a/tests/test_character_predictor.py b/tests/test_character_predictor.py
deleted file mode 100644
index 01bda78..0000000
--- a/tests/test_character_predictor.py
+++ /dev/null
@@ -1,31 +0,0 @@
-"""Test for CharacterPredictor class."""
-import importlib
-import os
-from pathlib import Path
-import unittest
-
-from loguru import logger
-
-from text_recognizer.character_predictor import CharacterPredictor
-from text_recognizer.networks import MLP
-
-SUPPORT_DIRNAME = Path(__file__).parents[0].resolve() / "support" / "emnist"
-
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
-
-
-class TestCharacterPredictor(unittest.TestCase):
- """Tests for the CharacterPredictor class."""
-
- def test_filename(self) -> None:
- """Test that CharacterPredictor correctly predicts on a single image, for serveral test images."""
- network_fn_ = MLP
- predictor = CharacterPredictor(network_fn=network_fn_)
-
- for filename in SUPPORT_DIRNAME.glob("*.png"):
- pred, conf = predictor.predict(str(filename))
- logger.info(
- f"Prediction: {pred} at confidence: {conf} for image with character {filename.stem}"
- )
- self.assertEqual(pred, filename.stem)
- self.assertGreater(conf, 0.7)