diff --git a/inkycal/tests/test_inkycal_stocks.py b/inkycal/tests/test_inkycal_stocks.py index e9ebaba..38a7b2e 100755 --- a/inkycal/tests/test_inkycal_stocks.py +++ b/inkycal/tests/test_inkycal_stocks.py @@ -1,56 +1,56 @@ -# #!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" -# } -# } -# ] -# -# -# 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)) -# -# -# if __name__ == '__main__': -# logger = logging.getLogger() -# logger.level = logging.DEBUG -# logger.addHandler(logging.StreamHandler(sys.stdout)) -# -# unittest.main() +#!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" + } + } +] + + +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)) + + +if __name__ == '__main__': + logger = logging.getLogger() + logger.level = logging.DEBUG + logger.addHandler(logging.StreamHandler(sys.stdout)) + + unittest.main()