test_OCR.py

import unittest
from OCR import OCR
from Screen import *


class OCRTestCase(unittest.TestCase):
    def test_simple_OCR(self):
        """ Simple check of OCR to bring back text of a know image. """
        ocr = OCR(screen=None)

        # Load image
        image_path = 'test/ocr/ocr-test1.png'
        orig_image = cv2.imread(image_path)
        ocr_textlist = ocr.image_simple_ocr(orig_image)
        actual = str(ocr_textlist)
        expected = "['DESTINATION', 'SIRIUS ATMOSPHERICS']"

        self.assertEqual(actual, expected)  # add assertion here

    def test_similarity_test1(self):
        ocr = OCR(screen=None)
        s1 = "Orbital Construction Site: Wingrove's Inheritance"
        s2 = "Wingrove's Inheritance (Orbital Construction Site)"
        actual = ocr.string_similarity(s1, s2)
        print(f"Dice: {actual}")

        self.assertGreater(actual, 0.9)  # add assertion here

    def test_similarity_test2(self):
        ocr = OCR(screen=None)
        s1 = "STAR BLAZE V2V-65W"
        s2 = "STAR BLAZE (V2V-65W)"
        actual = ocr.string_similarity(s1, s2)
        print(f"Dice: {actual}")

        self.assertGreater(actual, 0.8)  # add assertion here


if __name__ == '__main__':
    unittest.main()