Add 'labelImg/' from commit 'b33f965b6d14c14f1e46b247f1bf346e03f2e950'

git-subtree-dir: labelImg
git-subtree-mainline: dbe80aca78
git-subtree-split: b33f965b6d

labelImg/tests/.gitignore

@@ -0,0 +1 @@
+test.xml

labelImg/tests/test_io.py

@@ -0,0 +1,101 @@
+import os
+import sys
+import unittest
+
+class TestPascalVocRW(unittest.TestCase):
+
+    def test_upper(self):
+        dir_name = os.path.abspath(os.path.dirname(__file__))
+        libs_path = os.path.join(dir_name, '..', 'libs')
+        sys.path.insert(0, libs_path)
+        from pascal_voc_io import PascalVocWriter
+        from pascal_voc_io import PascalVocReader
+
+        # Test Write/Read
+        writer = PascalVocWriter('tests', 'test', (512, 512, 1), local_img_path='tests/test.512.512.bmp')
+        difficult = 1
+        writer.add_bnd_box(60, 40, 430, 504, 'person', difficult)
+        writer.add_bnd_box(113, 40, 450, 403, 'face', difficult)
+        writer.save('tests/test.xml')
+
+        reader = PascalVocReader('tests/test.xml')
+        shapes = reader.get_shapes()
+
+        person_bnd_box = shapes[0]
+        face = shapes[1]
+        self.assertEqual(person_bnd_box[0], 'person')
+        self.assertEqual(person_bnd_box[1], [(60, 40), (430, 40), (430, 504), (60, 504)])
+        self.assertEqual(face[0], 'face')
+        self.assertEqual(face[1], [(113, 40), (450, 40), (450, 403), (113, 403)])
+
+
+class TestCreateMLRW(unittest.TestCase):
+
+    def test_a_write(self):
+        dir_name = os.path.abspath(os.path.dirname(__file__))
+        libs_path = os.path.join(dir_name, '..', 'libs')
+        sys.path.insert(0, libs_path)
+        from create_ml_io import CreateMLWriter
+
+        person = {'label': 'person', 'points': ((65, 45), (420, 45), (420, 512), (65, 512))}
+        face = {'label': 'face', 'points': ((245, 250), (350, 250), (350, 365), (245, 365))}
+
+        expected_width = 105    # 350-245 -> create_ml_io.py ll 46
+        expected_height = 115   # 365-250 -> create_ml_io.py ll 49
+        expected_x = 297.5      # 245+105/2 -> create_ml_io.py ll 53
+        expected_y = 307.5      # 250+115/2 > create_ml_io.py ll 54
+
+        shapes = [person, face]
+        output_file = dir_name + "/tests.json"
+
+        writer = CreateMLWriter('tests', 'test.512.512.bmp', (512, 512, 1), shapes, output_file,
+                                local_img_path='tests/test.512.512.bmp')
+        
+        writer.verified = True
+        writer.write()
+
+        # check written json
+        with open(output_file, "r") as file:
+            input_data = file.read()
+
+        import json
+        data_dict = json.loads(input_data)[0]
+        self.assertEqual(True, data_dict['verified'], 'verified tag not reflected')
+        self.assertEqual('test.512.512.bmp', data_dict['image'], 'filename not correct in .json')
+        self.assertEqual(2, len(data_dict['annotations']), 'output file contains to less annotations')
+        face = data_dict['annotations'][1]
+        self.assertEqual('face', face['label'], 'label name is wrong')
+        face_coords = face['coordinates']
+        self.assertEqual(expected_width, face_coords['width'], 'calculated width is wrong')
+        self.assertEqual(expected_height, face_coords['height'], 'calculated height is wrong')
+        self.assertEqual(expected_x, face_coords['x'], 'calculated x is wrong')
+        self.assertEqual(expected_y, face_coords['y'], 'calculated y is wrong')
+
+    def test_b_read(self):
+        dir_name = os.path.abspath(os.path.dirname(__file__))
+        libs_path = os.path.join(dir_name, '..', 'libs')
+        sys.path.insert(0, libs_path)
+        from create_ml_io import CreateMLReader
+
+        output_file = dir_name + "/tests.json"
+        reader = CreateMLReader(output_file, 'tests/test.512.512.bmp')
+        shapes = reader.get_shapes()
+        face = shapes[1]
+
+        self.assertEqual(2, len(shapes), 'shape count is wrong')
+        self.assertEqual('face', face[0], 'label is wrong')
+
+        face_coords = face[1]
+        x_min = face_coords[0][0]
+        x_max = face_coords[1][0]
+        y_min = face_coords[0][1]
+        y_max = face_coords[2][1]
+
+        self.assertEqual(245, x_min, 'xmin is wrong')
+        self.assertEqual(350, x_max, 'xmax is wrong')
+        self.assertEqual(250, y_min, 'ymin is wrong')
+        self.assertEqual(365, y_max, 'ymax is wrong')
+
+
+if __name__ == '__main__':
+    unittest.main()

labelImg/tests/test_qt.py

@@ -0,0 +1,20 @@
+
+from unittest import TestCase
+
+from labelImg import get_main_app
+
+
+class TestMainWindow(TestCase):
+
+    app = None
+    win = None
+
+    def setUp(self):
+        self.app, self.win = get_main_app()
+
+    def tearDown(self):
+        self.win.close()
+        self.app.quit()
+
+    def test_noop(self):
+        pass

labelImg/tests/test_settings.py

@@ -0,0 +1,33 @@
+#!/usr/bin/env python
+import os
+import sys
+import time
+import unittest
+
+__author__ = 'TzuTaLin'
+
+dir_name = os.path.abspath(os.path.dirname(__file__))
+libs_path = os.path.join(dir_name, '..', 'libs')
+sys.path.insert(0, libs_path)
+from settings import Settings
+
+class TestSettings(unittest.TestCase):
+
+    def test_basic(self):
+        settings = Settings()
+        settings['test0'] = 'hello'
+        settings['test1'] = 10
+        settings['test2'] = [0, 2, 3]
+        self.assertEqual(settings.get('test3', 3), 3)
+        self.assertEqual(settings.save(), True)
+
+        settings.load()
+        self.assertEqual(settings.get('test0'), 'hello')
+        self.assertEqual(settings.get('test1'), 10)
+
+        settings.reset()
+        
+
+
+if __name__ == '__main__':
+    unittest.main()

labelImg/tests/test_stringBundle.py

@@ -0,0 +1,29 @@
+import os
+import sys
+import unittest
+import resources
+from stringBundle import StringBundle
+
+class TestStringBundle(unittest.TestCase):
+
+    def test_loadDefaultBundle_withoutError(self):
+        str_bundle = StringBundle.get_bundle('en')
+        self.assertEqual(str_bundle.get_string("openDir"), 'Open Dir', 'Fail to load the default bundle')
+
+    def test_fallback_withoutError(self):
+        str_bundle = StringBundle.get_bundle('zh-TW')
+        self.assertEqual(str_bundle.get_string("openDir"), u'\u958B\u555F\u76EE\u9304', 'Fail to load the zh-TW bundle')
+
+    def test_setInvaleLocaleToEnv_printErrorMsg(self):
+        prev_lc = os.environ['LC_ALL']
+        prev_lang = os.environ['LANG']
+        os.environ['LC_ALL'] = 'UTF-8'
+        os.environ['LANG'] = 'UTF-8'
+        str_bundle = StringBundle.get_bundle()
+        self.assertEqual(str_bundle.get_string("openDir"), 'Open Dir', 'Fail to load the default bundle')
+        os.environ['LC_ALL'] = prev_lc
+        os.environ['LANG'] = prev_lang
+
+
+if __name__ == '__main__':
+    unittest.main()

labelImg/tests/test_utils.py

@@ -0,0 +1,22 @@
+import os
+import sys
+import unittest
+from libs.utils import Struct, new_action, new_icon, add_actions, format_shortcut, generate_color_by_text, natural_sort
+
+class TestUtils(unittest.TestCase):
+
+    def test_generateColorByGivingUniceText_noError(self):
+        res = generate_color_by_text(u'\u958B\u555F\u76EE\u9304')
+        self.assertTrue(res.green() >= 0)
+        self.assertTrue(res.red() >= 0)
+        self.assertTrue(res.blue() >= 0)
+
+    def test_nautalSort_noError(self):
+        l1 = ['f1', 'f11', 'f3']
+        expected_l1 = ['f1', 'f3', 'f11']
+        natural_sort(l1)
+        for idx, val in enumerate(l1):
+            self.assertTrue(val == expected_l1[idx])
+
+if __name__ == '__main__':
+    unittest.main()