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