Update test_inkycal_stocks.py
This commit is contained in:
		| @@ -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() | ||||
|   | ||||
		Reference in New Issue
	
	Block a user