Fix #5 and add more tests (#7)

Author: vfdev-5

image_dataset_viz/dataset_exporter.py

@@ -25,7 +25,7 @@ def get_default_font(text_size):
     try:
         return ImageFont.truetype(font="DejaVuSans.ttf", size=text_size)
     except OSError:
-        font_path = Path(__file__).parent() / "fonts" / "DejaVuSans.ttf"
+        font_path = Path(__file__).parent / "fonts" / "DejaVuSans.ttf"
         return ImageFont.truetype(font=font_path.as_posix(), size=text_size)
 
 

tests/test_dataset_exporter.py

@@ -11,9 +11,10 @@
 from unittest import TestCase, main
 
 import numpy as np
-from PIL import Image
+from PIL import Image, ImageFont
 
-from image_dataset_viz.dataset_exporter import resize_image, DatasetExporter
+from image_dataset_viz.dataset_exporter import resize_image, DatasetExporter, \
+    get_default_font, to_pil, render_datapoint
 
 
 class TestDatasetExporter(TestCase):
@@ -24,6 +25,10 @@ def setUp(self):
     def tearDown(self):
         pass
 
+    def test_get_default_font(self):
+        font = get_default_font(10)
+        assert isinstance(font, ImageFont.FreeTypeFont)
+
     def test_resize_image(self):
         size = (320, 300)
         large_img = Image.new('RGB', size=size)
@@ -47,6 +52,20 @@ def read_img(i):
             de = DatasetExporter(read_img_fn=read_img)
             de.export_datapoint(0, 0, "test.png")
 
+    def test_render_datapoint(self):
+
+        img = np.ones((100, 120, 3), dtype=np.uint8)
+        res = render_datapoint(img, "test label", text_color=(0, 255, 0), text_size=10)
+        assert isinstance(res, Image.Image)
+
+        target = Image.fromarray(np.ones((100, 120, 3), dtype=np.uint8))
+        res = render_datapoint(img, target, text_color=(0, 255, 0), text_size=10)
+        assert isinstance(res, Image.Image)
+
+        target = np.array([[10, 10], [55, 10], [55, 77], [10, 77]])
+        res = render_datapoint(img, target, geom_color=(255, 0, 0))
+        assert isinstance(res, Image.Image)
+
     def test_export_datapoint(self):
 
         def read_img(i):
@@ -128,6 +147,15 @@ def read_img(i):
                 out_img = Image.open(fp)
                 self.assertEqual(out_img.size, ((s + m) * n_cols, (s + m) * max_n_rows))
 
+    def test_to_pil(self):
+        img = np.ones((100, 120, 3), dtype=np.uint8)
+        pil_img = to_pil(img)
+        assert isinstance(pil_img, Image.Image)
+
+        img = Image.fromarray(np.ones((100, 120, 3), dtype=np.uint8))
+        pil_img = to_pil(img)
+        assert isinstance(pil_img, Image.Image)
+
 
 if __name__ == "__main__":
     main()