Update test_inkycal_stocks.py
This commit is contained in:
		| @@ -1,56 +1,75 @@ | |||||||
| #!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 = [ | ||||||
|     { | { | ||||||
|         "name": "Stocks", |   "position": 1, | ||||||
|         "config": { |   "name": "Stocks", | ||||||
|             "size": [528, 30], |   "config": { | ||||||
|             "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'], |       "size": [528, 20], | ||||||
|             "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en" |       "tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'], | ||||||
|         } |       "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en" | ||||||
|     }, |       } | ||||||
|     { | }, | ||||||
|         "name": "Stocks", | { | ||||||
|         "config": { |   "position": 1, | ||||||
|             "size": [528, 50], |   "name": "Stocks", | ||||||
|             "tickers": [], |   "config": { | ||||||
|             "padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en" |       "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): | 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 = "") | ||||||
|         Module.get_config() |     Module.get_config() | ||||||
|         print('OK') |     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_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__': | if __name__ == '__main__': | ||||||
|     logger = logging.getLogger() |   unittest.main() | ||||||
|     logger.level = logging.DEBUG |  | ||||||
|     logger.addHandler(logging.StreamHandler(sys.stdout)) |  | ||||||
|  |  | ||||||
|     unittest.main() |  | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user