Add test for edge case

Author: Belval

TextRecognitionDataGenerator/computer_text_generator.py

@@ -10,7 +10,7 @@ def generate(cls, text, font, text_color, font_size, orientation, space_width):
         elif orientation == 1:
             return cls.__generate_vertical_text(text, font, text_color, font_size, space_width)
         else:
-            raise Exception("Unknown orientation " + str(orientation))
+            raise ValueError("Unknown orientation " + str(orientation))
 
     @classmethod
     def __generate_horizontal_text(cls, text, font, text_color, font_size, space_width):

tests.py

@@ -453,6 +453,34 @@ def test_generate_vertical_text_with_variable_space(self):
 
         os.remove('tests/out/TEST TEST TEST_12.jpg')
 
+    def test_generate_text_with_unknown_orientation(self):
+        try:
+            FakeTextDataGenerator.generate(
+                12,
+                'TEST TEST TEST',
+                'tests/font.ttf',
+                'tests/out/',
+                32,
+                'jpg',
+                0,
+                False,
+                0,
+                False,
+                1,
+                0,
+                0,
+                False,
+                0,
+                -1,
+                0,
+                '#010101',
+                100,
+                2
+            )
+            raise Exception("Unknown orientation did not throw")
+        except ValueError:
+            pass
+
     def test_generate_string_with_letters(self):
         s = create_strings_randomly(1, False, 1, True, False, False, 'en')[0]