Update test_inkycal_stocks.py

aceinnolab · Nov 20, 2023 · 341a9cd · 341a9cd
1 parent a937564
commit 341a9cd
Showing 1 changed file with 65 additions and 46 deletions.
diff --git a/inkycal/tests/test_inkycal_stocks.py b/inkycal/tests/test_inkycal_stocks.py
@@ -1,56 +1,75 @@
-#!python3
-"""
-inkycal_stocks unittest
-"""
-import logging
-import sys
 import unittest
 from inkycal.modules import Stocks as Module
 
-from inkycal.modules.inky_image import Inkyimage
-from inkycal.tests import Config
-preview = Inkyimage.preview
-merge = Inkyimage.merge
-
 tests = [
-    {
-        "name": "Stocks",
-        "config": {
-            "size": [528, 30],
-            "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'],
-            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
-        }
-    },
-    {
-        "name": "Stocks",
-        "config": {
-            "size": [528, 50],
-            "tickers": [],
-            "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
-        }
-    }
+{
+  "position": 1,
+  "name": "Stocks",
+  "config": {
+      "size": [528, 20],
+      "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],
+      "tickers": [],
+      "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
+      }
+},
+{
+  "position": 1,
+  "name": "Stocks",
+  "config": {
+      "size": [528, 200],
+      "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, 800],
+      "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, 100],
+      "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, 400],
+      "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'],
+      "padding_x": 10, "padding_y": 10, "fontsize": 14, "language": "en"
+      }
+},
 ]
 
-
 class module_test(unittest.TestCase):
-    def test_get_config(self):
-        print('getting data for web-ui...', end="")
-        Module.get_config()
-        print('OK')
-
-    def test_generate_image(self):
-        for test in tests:
-            print(f'test {tests.index(test) + 1} generating image..')
-            module = Module(test)
-            im_black, im_colour = module.generate_image()
-            print('OK')
-            if Config.USE_PREVIEW:
-                preview(merge(im_black, im_colour))
+  def test_get_config(self):
+    print('getting data for web-ui...', end = "")
+    Module.get_config()
+    print('OK')
 
+  def test_generate_image(self):
+    for test in tests:
+      print(f'test {tests.index(test)+1} generating image..')
+      module = Module(test)
+      module.generate_image()
+      print('OK')
 
 if __name__ == '__main__':
-    logger = logging.getLogger()
-    logger.level = logging.DEBUG
-    logger.addHandler(logging.StreamHandler(sys.stdout))
-
-    unittest.main()
+  unittest.main()