From 24c7d4c277402524f15c703d021c00fc81d53956 Mon Sep 17 00:00:00 2001
From: Robert Sachunsky <sachunsky@informatik.uni-leipzig.de>
Date: Wed, 3 Jun 2026 20:58:05 +0200
Subject: [PATCH] update trocr smoke test, add cnnrnn ocr smoke test

---
 tests/test_model_zoo.py | 24 ++++++++++++++++++------
 1 file changed, 18 insertions(+), 6 deletions(-)

diff --git a/tests/test_model_zoo.py b/tests/test_model_zoo.py
index 341bc21..2902bfe 100644
--- a/tests/test_model_zoo.py
+++ b/tests/test_model_zoo.py
@@ -1,16 +1,28 @@
 from eynollah.model_zoo import EynollahModelZoo
+from eynollah.predictor import Predictor
 
 def test_trocr1(
     model_dir,
 ):
     model_zoo = EynollahModelZoo(model_dir)
     try:
-        from transformers import TrOCRProcessor, VisionEncoderDecoderModel
-        model_zoo.load_models('trocr_processor',
-                              ('ocr', 'tr'))
-        proc = model_zoo.get('trocr_processor')
-        assert isinstance(proc, TrOCRProcessor)
+        model_zoo.load_models(('ocr', 'tr'))
         model = model_zoo.get('ocr')
-        assert isinstance(model, VisionEncoderDecoderModel)
+        assert isinstance(model, Predictor)
+        shape = model.input_shape
+        assert len(shape) == 3
+    except ImportError:
+        pass
+
+def test_cnnrnnocr1(
+    model_dir,
+):
+    model_zoo = EynollahModelZoo(model_dir)
+    try:
+        model_zoo.load_models('ocr')
+        model = model_zoo.get('ocr')
+        assert isinstance(model, Predictor)
+        shape = model.input_shape
+        assert len(shape) == 4
     except ImportError:
         pass