# Copyright (c) 2021-2025  The University of Texas Southwestern Medical Center.
# All rights reserved.

# Redistribution and use in source and binary forms, with or without
# modification, are permitted for academic and research use only
# (subject to the limitations in the disclaimer below)
# provided that the following conditions are met:

#      * Redistributions of source code must retain the above copyright notice,
#      this list of conditions and the following disclaimer.

#      * Redistributions in binary form must reproduce the above copyright
#      notice, this list of conditions and the following disclaimer in the
#      documentation and/or other materials provided with the distribution.

#      * Neither the name of the copyright holders nor the names of its
#      contributors may be used to endorse or promote products derived from this
#      software without specific prior written permission.

# NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE GRANTED BY
# THIS LICENSE. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
# CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
# PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
# BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
# IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.

# Standard Library Imports
import unittest

# Third-Party Imports
import numpy as np
from PIL import Image, ImageFont

# import pytest

# Local Imports
from navigate.tools.image import text_array, create_arrow_image


class TextArrayTestCase(unittest.TestCase):
    def test_text_array(self):
        text = "55"
        offset = (0, 0)

        result = text_array(text, offset)

        # Assert that the result is a numpy array
        self.assertIsInstance(result, np.ndarray)

        # Assert that the values in the result array are either True or False
        assert result.dtype is np.dtype("bool")

    def test_text_array_output_type(self):
        """Confirm output is np.ndarray object"""
        text_output = text_array(text="Navigate")
        assert type(text_output) == np.ndarray

    def test_text_array_output_height(self):
        """Confirm that the output is approximately the correct height

        Initially thought that the height should be ~= font_size, but this
        turned out to be much more variable than I anticipated
        """
        text = "Navigate"
        text_output = text_array(text=text)
        font = ImageFont.load_default()  # match font size in text_array()
        height = np.shape(text_output)[0]
        width = np.shape(text_output)[1]
        _, _, expected_width, expected_height = font.getbbox(text)
        assert width == expected_width
        assert height == expected_height


class TestCreateArrowImage(unittest.TestCase):
    def test_create_arrow_image(self):
        xys = [(50, 50), (150, 50), (200, 100)]
        image = create_arrow_image(xys, direction="right")
        self.assertIsInstance(image, Image.Image)
        self.assertEqual(image.width, 300)
        self.assertEqual(image.height, 200)

        xys = [(50, 50), (150, 50), (200, 100)]
        image = create_arrow_image(xys, 400, 300, direction="left")
        self.assertIsInstance(image, Image.Image)
        self.assertEqual(image.width, 400)
        self.assertEqual(image.height, 300)

        image2 = create_arrow_image(xys, 500, 400, direction="up", image=image)
        self.assertIsInstance(image2, Image.Image)
        self.assertEqual(image2.width, 400)
        self.assertEqual(image2.height, 300)
        assert image == image2

        image3 = create_arrow_image(xys, 500, 400, direction="down", image=image)
        self.assertIsInstance(image3, Image.Image)
        self.assertEqual(image3.width, 400)
        self.assertEqual(image3.height, 300)
        assert image == image3


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