iicd/Inkycal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
aceisace
2022-04-14 06:07:14 +02:00
parent 40c0f5ff58
commit ef678a970a
2 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
inkycal/main.py
View File

@@ -6,16 +6,14 @@ Main class for inkycal Project
Copyright by aceisace Copyright by aceisace
""" """
import os
import traceback
import arrow
import time
import json import json
import logging import traceback
from logging.handlers import RotatingFileHandler from logging.handlers import RotatingFileHandler
from inkycal.display import Display import arrow
from inkycal.custom import * from inkycal.custom import *
from inkycal.display import Display
from inkycal.modules.inky_image import Inkyimage as Images from inkycal.modules.inky_image import Inkyimage as Images
try: try:
@@ -165,7 +163,7 @@ class Inkycal:
print(str(e)) print(str(e))
# Path to store images # Path to store images
self.image_folder = top_level + '/images' self.image_folder = image_folder
# Give an OK message # Give an OK message
print('loaded inkycal') print('loaded inkycal')
@@ -219,8 +217,8 @@ class Inkycal:
print(f'generating image(s) for {name}...', end="") print(f'generating image(s) for {name}...', end="")
try: try:
black, colour = module.generate_image() black, colour = module.generate_image()
black.save(f"{self.image_folder}/module{number}_black.png", "PNG") black.save(f"{self.image_folder}module{number}_black.png", "PNG")
colour.save(f"{self.image_folder}/module{number}_colour.png", "PNG") colour.save(f"{self.image_folder}module{number}_colour.png", "PNG")
print('OK!') print('OK!')
except: except:
errors.append(number) errors.append(number)
@@ -334,7 +332,7 @@ class Inkycal:
returns the merged image returns the merged image
""" """
im1_path, im2_path = images + 'canvas.png', images + 'canvas_colour.png' im1_path, im2_path = image_folder + 'canvas.png', image_folder + 'canvas_colour.png'
# If there is an image for black and colour, merge them # If there is an image for black and colour, merge them
if os.path.exists(im1_path) and os.path.exists(im2_path): if os.path.exists(im1_path) and os.path.exists(im2_path):
@@ -446,7 +444,7 @@ class Inkycal:
im_black.save(self.image_folder + '/canvas.png', 'PNG') im_black.save(self.image_folder + '/canvas.png', 'PNG')
im_colour.save(self.image_folder + '/canvas_colour.png', 'PNG') im_colour.save(self.image_folder + '/canvas_colour.png', 'PNG')
# Additionally combine the two images with color # Additionally, combine the two images with color
def clear_white(img): def clear_white(img):
"""Replace all white pixels from image with transparent pixels """Replace all white pixels from image with transparent pixels
""" """
@@ -454,7 +452,7 @@ class Inkycal:
x[:, :, 3] = (255 * (x[:, :, :3] != 255).any(axis=2)).astype(numpy.uint8) x[:, :, 3] = (255 * (x[:, :, :3] != 255).any(axis=2)).astype(numpy.uint8)
return Image.fromarray(x) return Image.fromarray(x)
# Additionally combine the two images with color # Additionally, combine the two images with color
def black_to_colour(img): def black_to_colour(img):
"""Replace all black pixels from image with red pixels """Replace all black pixels from image with red pixels
""" """
@@ -473,7 +471,7 @@ class Inkycal:
im_colour = black_to_colour(im_colour) im_colour = black_to_colour(im_colour)
im_colour.paste(im_black, (0, 0), im_black) im_colour.paste(im_black, (0, 0), im_black)
im_colour.save(images + 'full-screen.png', 'PNG') im_colour.save(image_folder + 'full-screen.png', 'PNG')
@staticmethod @staticmethod
def _optimize_im(image, threshold=220): def _optimize_im(image, threshold=220):

25
inkycal/tests/main_test.py
View File

@@ -55,18 +55,21 @@ test_config = """
} }
""" """
class module_test(unittest.TestCase):
def test_without_rendering(self):
# Create temporary json settings file with the config from above
with open('dummy.json', mode="w") as file:
file.write(test_config)
print('testing Inkycal in non-render-mode...', end = "")
inky = Inkycal('dummy.json', render=False)
inky.test()
print('OK')
os.remove('dummy.json')
class ModuleTest(unittest.TestCase):
@staticmethod
def test_without_rendering():
# Create temporary json settings file with the config from above
with open('dummy.json', mode="w") as file:
file.write(test_config)
print('testing Inkycal in non-render-mode...', end="")
inky = Inkycal('dummy.json', render=False)
inky.test()
print('OK')
os.remove('dummy.json')
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()