Improved testing method

Added shebangs, logging messages are now shown.
Running on an Rpi allows previewing the generated images.

inkycal/tests/inkycal_stocks_test.py

@@ -1,27 +1,38 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+
+"""
+Stocks test (inkycal_stocks)
+Copyright by aceisace
+"""
+
 import unittest
 from inkycal.modules import Stocks as Module
+from helper_functions import *
+environment = get_environment()
+
+# Set to True to preview images. Only works on Raspberry Pi OS with Desktop
+use_preview = False
+
 
 tests = [
 {
-  "position": 1,
   "name": "Stocks",
   "config": {
-      "size": [528, 20],
+      "size": [528, 30],
       "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'],
       "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
       }
 },
 {
-  "position": 1,
   "name": "Stocks",
   "config": {
-      "size": [528, 20],
+      "size": [528, 50],
       "tickers": [],
       "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
       }
 },
 {
-  "position": 1,
   "name": "Stocks",
   "config": {
       "size": [528, 200],
@@ -30,7 +41,6 @@ tests = [
       }
 },
 {
-  "position": 1,
   "name": "Stocks",
   "config": {
       "size": [528, 800],
@@ -39,7 +49,6 @@ tests = [
       }
 },
 {
-  "position": 1,
   "name": "Stocks",
   "config": {
       "size": [528, 100],
@@ -48,7 +57,6 @@ tests = [
       }
 },
 {
-  "position": 1,
   "name": "Stocks",
   "config": {
       "size": [528, 400],
@@ -68,8 +76,15 @@ class module_test(unittest.TestCase):
     for test in tests:
       print(f'test {tests.index(test)+1} generating image..')
       module = Module(test)
-      module.generate_image()
+      im_black, im_colour = module.generate_image()
       print('OK')
+      if use_preview == True and environment == 'Raspberry':
+        preview(merge(im_black, im_colour))
 
 if __name__ == '__main__':
+
+  logger = logging.getLogger()
+  logger.level = logging.DEBUG
+  logger.addHandler(logging.StreamHandler(sys.stdout))
+
   unittest.main()