58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
|
import logging
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from inkycal.modules import Stocks
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
|
|
||
|
|
tests = [
|
||
|
|
{
|
||
|
|
"position": 1,
|
||
|
|
"name": "Stocks",
|
||
|
|
"config": {
|
||
|
|
"size": [400, 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": [400, 200],
|
||
|
|
"tickers": [],
|
||
|
|
"padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"position": 1,
|
||
|
|
"name": "Stocks",
|
||
|
|
"config": {
|
||
|
|
"size": [400, 300],
|
||
|
|
"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": [400, 400],
|
||
|
|
"tickers": ['TSLA', 'AMD', 'NVDA', '^DJI', 'BTC-USD', 'EURUSD=X'],
|
||
|
|
"padding_x": 10, "padding_y": 10, "fontsize": 12, "language": "en"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class TestStocks(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_generate_image(self):
|
||
|
|
for test in tests:
|
||
|
|
logger.info(f'test {tests.index(test) + 1} generating image..')
|
||
|
|
module = Stocks(test)
|
||
|
|
module.generate_image()
|
||
|
|
logger.info('OK')
|
||
|
|
|