Update test_inkycal_stocks.py
This commit is contained in:
		| @@ -1,37 +1,63 @@ | |||||||
| #!python3 |  | ||||||
| """ |  | ||||||
| inkycal_stocks unittest |  | ||||||
| """ |  | ||||||
| import logging |  | ||||||
| import sys |  | ||||||
| import unittest | import unittest | ||||||
| from inkycal.modules import Stocks as Module | 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 = [ | tests = [ | ||||||
| { | { | ||||||
|  |   "position": 1, | ||||||
|   "name": "Stocks", |   "name": "Stocks", | ||||||
|   "config": { |   "config": { | ||||||
|             "size": [528, 30], |       "size": [528, 20], | ||||||
|       "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'], |       "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'], | ||||||
|       "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en" |       "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en" | ||||||
|       } |       } | ||||||
| }, | }, | ||||||
| { | { | ||||||
|  |   "position": 1, | ||||||
|   "name": "Stocks", |   "name": "Stocks", | ||||||
|   "config": { |   "config": { | ||||||
|             "size": [528, 50], |       "size": [528, 20], | ||||||
|       "tickers": [], |       "tickers": [], | ||||||
|       "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en" |       "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): | class module_test(unittest.TestCase): | ||||||
|   def test_get_config(self): |   def test_get_config(self): | ||||||
|     print('getting data for web-ui...', end = "") |     print('getting data for web-ui...', end = "") | ||||||
| @@ -42,15 +68,8 @@ class module_test(unittest.TestCase): | |||||||
|     for test in tests: |     for test in tests: | ||||||
|       print(f'test {tests.index(test)+1} generating image..') |       print(f'test {tests.index(test)+1} generating image..') | ||||||
|       module = Module(test) |       module = Module(test) | ||||||
|             im_black, im_colour = module.generate_image() |       module.generate_image() | ||||||
|       print('OK') |       print('OK') | ||||||
|             if Config.USE_PREVIEW: |  | ||||||
|                 preview(merge(im_black, im_colour)) |  | ||||||
|  |  | ||||||
|  |  | ||||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||||
|     logger = logging.getLogger() |  | ||||||
|     logger.level = logging.DEBUG |  | ||||||
|     logger.addHandler(logging.StreamHandler(sys.stdout)) |  | ||||||
|  |  | ||||||
|   unittest.main() |   unittest.main() | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user