645198d58b
Current Linux distros ship version 5 of the tesseract OCR software, so the nextcube screen test is ignored there. Let's make the check more flexible to allow newer versions, too, and remove the old v3 test since most Linux distros don't ship this version anymore. Message-ID: <20231101204323.35533-1-huth@tuxfamily.org> Reviewed-by: Philippe Mathieu-Daudé <philmd@linaro.org> Signed-off-by: Thomas Huth <huth@tuxfamily.org>
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
# Functional test that boots a VM and run OCR on the framebuffer
|
|
#
|
|
# Copyright (c) 2019 Philippe Mathieu-Daudé <f4bug@amsat.org>
|
|
#
|
|
# This work is licensed under the terms of the GNU GPL, version 2 or
|
|
# later. See the COPYING file in the top-level directory.
|
|
|
|
import os
|
|
import time
|
|
|
|
from avocado_qemu import QemuSystemTest
|
|
from avocado import skipUnless
|
|
|
|
from tesseract_utils import tesseract_available, tesseract_ocr
|
|
|
|
PIL_AVAILABLE = True
|
|
try:
|
|
from PIL import Image
|
|
except ImportError:
|
|
PIL_AVAILABLE = False
|
|
|
|
|
|
class NextCubeMachine(QemuSystemTest):
|
|
"""
|
|
:avocado: tags=arch:m68k
|
|
:avocado: tags=machine:next-cube
|
|
:avocado: tags=device:framebuffer
|
|
"""
|
|
|
|
timeout = 15
|
|
|
|
def check_bootrom_framebuffer(self, screenshot_path):
|
|
rom_url = ('https://sourceforge.net/p/previous/code/1350/tree/'
|
|
'trunk/src/Rev_2.5_v66.BIN?format=raw')
|
|
rom_hash = 'b3534796abae238a0111299fc406a9349f7fee24'
|
|
rom_path = self.fetch_asset(rom_url, asset_hash=rom_hash)
|
|
|
|
self.vm.add_args('-bios', rom_path)
|
|
self.vm.launch()
|
|
|
|
self.log.info('VM launched, waiting for display')
|
|
# TODO: Use avocado.utils.wait.wait_for to catch the
|
|
# 'displaysurface_create 1120x832' trace-event.
|
|
time.sleep(2)
|
|
|
|
self.vm.cmd('human-monitor-command',
|
|
command_line='screendump %s' % screenshot_path)
|
|
|
|
@skipUnless(PIL_AVAILABLE, 'Python PIL not installed')
|
|
def test_bootrom_framebuffer_size(self):
|
|
screenshot_path = os.path.join(self.workdir, "dump.ppm")
|
|
self.check_bootrom_framebuffer(screenshot_path)
|
|
|
|
width, height = Image.open(screenshot_path).size
|
|
self.assertEqual(width, 1120)
|
|
self.assertEqual(height, 832)
|
|
|
|
# Tesseract 4 adds a new OCR engine based on LSTM neural networks. The
|
|
# new version is faster and more accurate than version 3. The drawback is
|
|
# that it is still alpha-level software.
|
|
@skipUnless(tesseract_available(4), 'tesseract OCR tool not available')
|
|
def test_bootrom_framebuffer_ocr_with_tesseract(self):
|
|
screenshot_path = os.path.join(self.workdir, "dump.ppm")
|
|
self.check_bootrom_framebuffer(screenshot_path)
|
|
lines = tesseract_ocr(screenshot_path, tesseract_version=4)
|
|
text = '\n'.join(lines)
|
|
self.assertIn('Testing the FPU', text)
|
|
self.assertIn('System test failed. Error code', text)
|
|
self.assertIn('Boot command', text)
|
|
self.assertIn('Next>', text)
|