tests best practices

2023-11-21 15:18:19 +01:00
parent 96a972e31f
commit 59c59e80f5
18 changed files with 177 additions and 304 deletions
--- a/tests/test_inkycal_image.py
+++ b/tests/test_inkycal_image.py
@@ -0,0 +1,116 @@
+"""
+inkycal_image unittest
+"""
+import logging
+import unittest
+
+import requests
+from PIL import Image
+
+from inkycal.modules import Inkyimage as Module
+from inkycal.modules.inky_image import Inkyimage
+from tests import Config
+
+preview = Inkyimage.preview
+merge = Inkyimage.merge
+
+url = "https://github.com/aceinnolab/Inkycal/raw/assets/Repo/coffee.png"
+
+im = Image.open(requests.get(url, stream=True).raw)
+im.save("test.png", "PNG")
+test_path = "test.png"
+
+logger = logging.getLogger(__name__)
+logging.basicConfig(level=logging.DEBUG)
+
+tests = [
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [400, 200],
+            "path": test_path,
+            "palette": "bwr",
+            "autoflip": True,
+            "orientation": "vertical",
+            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
+        }
+    },
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [800, 500],
+            "path": test_path,
+            "palette": "bwy",
+            "autoflip": True,
+            "orientation": "vertical",
+            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
+        }
+    },
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [400, 100],
+            "path": test_path,
+            "palette": "bw",
+            "autoflip": False,
+            "orientation": "vertical",
+            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
+        }
+    },
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [400, 100],
+            "path": test_path,
+            "palette": "bwr",
+            "autoflip": True,
+            "orientation": "vertical",
+            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
+        }
+    },
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [400, 100],
+            "path": test_path,
+            "palette": "bwy",
+            "autoflip": True,
+            "orientation": "horizontal",
+            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
+        }
+    },
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [500, 800],
+            "path": test_path,
+            "palette": "bw",
+            "autoflip": True,
+            "orientation": "vertical",
+            "padding_x": 0, "padding_y": 0, "fontsize": 12, "language": "en"
+        }
+    },
+    {
+        "name": "Inkyimage",
+        "config": {
+            "size": [500, 800],
+            "path": test_path,
+            "palette": "bwr",
+            "autoflip": True,
+            "orientation": "vertical",
+            "padding_x": 20, "padding_y": 20, "fontsize": 12, "language": "en"
+        }
+    },
+]
+
+
+class TestInkyImage(unittest.TestCase):
+
+    def test_generate_image(self):
+        for test in tests:
+            logger.info(f'test {tests.index(test) + 1} generating image..')
+            module = Module(test)
+            im_black, im_colour = module.generate_image()
+            logger.info('OK')
+            if Config.USE_PREVIEW:
+                preview(merge(im_black, im_colour))