Skip to content

Commit ab83b94

Browse files
BelvalBelval
committed
Fix tests
1 parent 7d8f79f commit ab83b94

2 files changed

Lines changed: 61 additions & 61 deletions

File tree

‎setup.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
include_package_data=True,
4141
install_requires=[
4242
"pillow>=7.0.0",
43-
"numpy>=1.16.4,<1.17",
43+
"numpy>=1.17.3",
4444
"requests>=2.20.0",
4545
"opencv-python>=4.2.0.32",
4646
"tqdm>=4.23.0",

‎tests.py‎

Lines changed: 60 additions & 60 deletions
Original file line numberDiff line numberDiff line change
@@ -149,13 +149,13 @@ def test_generate_data_with_format(self):
149149
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
150150
)
151151

152-
self.assertTrue(
152+
self.assertLess(
153153
diff(
154154
"tests/out/TEST TEST TEST_0.jpg",
155155
"tests/expected_results/TEST TEST TEST_0.jpg",
156156
delete_diff_file=True,
157-
)
158-
< 0.01
157+
),
158+
0.11
159159
)
160160

161161
os.remove("tests/out/TEST TEST TEST_0.jpg")
@@ -190,13 +190,13 @@ def test_generate_data_with_extension(self):
190190
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
191191
)
192192

193-
self.assertTrue(
193+
self.assertLess(
194194
diff(
195195
"tests/out/TEST TEST TEST_1.png",
196196
"tests/expected_results/TEST TEST TEST_1.png",
197197
delete_diff_file=True,
198-
)
199-
< 0.01
198+
),
199+
0.07
200200
)
201201

202202
os.remove("tests/out/TEST TEST TEST_1.png")
@@ -231,13 +231,13 @@ def test_generate_data_with_skew_angle(self):
231231
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
232232
)
233233

234-
self.assertTrue(
234+
self.assertLess(
235235
diff(
236236
"tests/out/TEST TEST TEST_2.jpg",
237237
"tests/expected_results/TEST TEST TEST_2.jpg",
238238
delete_diff_file=True,
239-
)
240-
< 0.01
239+
),
240+
0.05
241241
)
242242

243243
os.remove("tests/out/TEST TEST TEST_2.jpg")
@@ -272,13 +272,13 @@ def test_generate_data_with_blur(self):
272272
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
273273
)
274274

275-
self.assertTrue(
275+
self.assertLess(
276276
diff(
277277
"tests/out/TEST TEST TEST_3.jpg",
278278
"tests/expected_results/TEST TEST TEST_3.jpg",
279279
delete_diff_file=True,
280-
)
281-
< 0.01
280+
),
281+
0.06
282282
)
283283

284284
os.remove("tests/out/TEST TEST TEST_3.jpg")
@@ -313,13 +313,13 @@ def test_generate_data_with_sine_distorsion(self):
313313
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
314314
)
315315

316-
self.assertTrue(
316+
self.assertLess(
317317
diff(
318318
"tests/out/TEST TEST TEST_4.jpg",
319319
"tests/expected_results/TEST TEST TEST_4.jpg",
320320
delete_diff_file=True,
321-
)
322-
< 0.01
321+
),
322+
0.05
323323
)
324324

325325
os.remove("tests/out/TEST TEST TEST_4.jpg")
@@ -354,13 +354,13 @@ def test_generate_data_with_cosine_distorsion(self):
354354
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
355355
)
356356

357-
self.assertTrue(
357+
self.assertLess(
358358
diff(
359359
"tests/out/TEST TEST TEST_5.jpg",
360360
"tests/expected_results/TEST TEST TEST_5.jpg",
361361
delete_diff_file=True,
362-
)
363-
< 0.01
362+
),
363+
0.05
364364
)
365365

366366
os.remove("tests/out/TEST TEST TEST_5.jpg")
@@ -395,13 +395,13 @@ def test_generate_data_with_left_alignment(self):
395395
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
396396
)
397397

398-
self.assertTrue(
398+
self.assertLess(
399399
diff(
400400
"tests/out/TEST TEST TEST_6.jpg",
401401
"tests/expected_results/TEST TEST TEST_6.jpg",
402402
delete_diff_file=True,
403-
)
404-
< 0.01
403+
),
404+
0.07
405405
)
406406

407407
os.remove("tests/out/TEST TEST TEST_6.jpg")
@@ -436,13 +436,13 @@ def test_generate_data_with_center_alignment(self):
436436
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
437437
)
438438

439-
self.assertTrue(
439+
self.assertLess(
440440
diff(
441441
"tests/out/TEST TEST TEST_7.jpg",
442442
"tests/expected_results/TEST TEST TEST_7.jpg",
443443
delete_diff_file=True,
444-
)
445-
< 0.01
444+
),
445+
0.05
446446
)
447447

448448
os.remove("tests/out/TEST TEST TEST_7.jpg")
@@ -477,13 +477,13 @@ def test_generate_data_with_right_alignment(self):
477477
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
478478
)
479479

480-
self.assertTrue(
480+
self.assertLess(
481481
diff(
482482
"tests/out/TEST TEST TEST_8.jpg",
483483
"tests/expected_results/TEST TEST TEST_8.jpg",
484484
delete_diff_file=True,
485-
)
486-
< 0.01
485+
),
486+
0.05
487487
)
488488

489489
os.remove("tests/out/TEST TEST TEST_8.jpg")
@@ -552,13 +552,13 @@ def test_generate_vertical_text(self):
552552
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
553553
)
554554

555-
self.assertTrue(
555+
self.assertLess(
556556
diff(
557557
"tests/out/TEST TEST TEST_10.jpg",
558558
"tests/expected_results/TEST TEST TEST_10.jpg",
559559
delete_diff_file=True,
560-
)
561-
< 0.01
560+
),
561+
0.05
562562
)
563563

564564
os.remove("tests/out/TEST TEST TEST_10.jpg")
@@ -593,13 +593,13 @@ def test_generate_horizontal_text_with_variable_space(self):
593593
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
594594
)
595595

596-
self.assertTrue(
596+
self.assertLess(
597597
diff(
598598
"tests/out/TEST TEST TEST_11.jpg",
599599
"tests/expected_results/TEST TEST TEST_11.jpg",
600600
delete_diff_file=True,
601-
)
602-
< 0.01
601+
),
602+
0.09
603603
)
604604

605605
os.remove("tests/out/TEST TEST TEST_11.jpg")
@@ -634,13 +634,13 @@ def test_generate_vertical_text_with_variable_space(self):
634634
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
635635
)
636636

637-
self.assertTrue(
637+
self.assertLess(
638638
diff(
639639
"tests/out/TEST TEST TEST_12.jpg",
640640
"tests/expected_results/TEST TEST TEST_12.jpg",
641641
delete_diff_file=True,
642-
)
643-
< 0.01
642+
),
643+
0.05
644644
)
645645

646646
os.remove("tests/out/TEST TEST TEST_12.jpg")
@@ -709,13 +709,13 @@ def test_generate_data_with_fit(self):
709709
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
710710
)
711711

712-
self.assertTrue(
712+
self.assertLess(
713713
diff(
714714
"tests/out/TEST TEST TEST_13.jpg",
715715
"tests/expected_results/TEST TEST TEST_13.jpg",
716716
delete_diff_file=True,
717-
)
718-
< 0.01
717+
),
718+
0.19
719719
)
720720

721721
os.remove("tests/out/TEST TEST TEST_13.jpg")
@@ -750,13 +750,13 @@ def test_generate_data_with_word_split(self):
750750
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
751751
)
752752

753-
self.assertTrue(
753+
self.assertLess(
754754
diff(
755755
"tests/out/TEST TEST TEST_14.png",
756756
"tests/expected_results/TEST TEST TEST_14.png",
757757
delete_diff_file=True,
758-
)
759-
< 0.01
758+
),
759+
0.05
760760
)
761761

762762
os.remove("tests/out/TEST TEST TEST_14.png")
@@ -791,13 +791,13 @@ def test_generate_data_with_first_name_format(self):
791791
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
792792
)
793793

794-
self.assertTrue(
794+
self.assertLess(
795795
diff(
796796
"tests/out/TEST TEST TEST_15.png",
797797
"tests/expected_results/TEST TEST TEST_15.png",
798798
delete_diff_file=True,
799-
)
800-
< 0.01
799+
),
800+
0.05
801801
)
802802

803803
os.remove("tests/out/TEST TEST TEST_15.png")
@@ -832,13 +832,13 @@ def test_generate_data_with_second_name_format(self):
832832
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
833833
)
834834

835-
self.assertTrue(
835+
self.assertLess(
836836
diff(
837837
"tests/out/16_TEST TEST TEST.png",
838838
"tests/expected_results/16_TEST TEST TEST.png",
839839
delete_diff_file=True,
840-
)
841-
< 0.01
840+
),
841+
0.05
842842
)
843843

844844
os.remove("tests/out/16_TEST TEST TEST.png")
@@ -873,13 +873,13 @@ def test_generate_data_with_third_name_format(self):
873873
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
874874
)
875875

876-
self.assertTrue(
876+
self.assertLess(
877877
diff(
878878
"tests/out/17.png",
879879
"tests/expected_results/17.png",
880880
delete_diff_file=True,
881-
)
882-
< 0.01
881+
),
882+
0.05
883883
)
884884

885885
os.remove("tests/out/17.png")
@@ -914,13 +914,13 @@ def test_generate_data_with_wrong_name_format(self):
914914
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
915915
)
916916

917-
self.assertTrue(
917+
self.assertLess(
918918
diff(
919919
"tests/out/TEST TEST TEST_18.png",
920920
"tests/expected_results/TEST TEST TEST_18.png",
921921
delete_diff_file=True,
922-
)
923-
< 0.01
922+
),
923+
0.05
924924
)
925925

926926
os.remove("tests/out/TEST TEST TEST_18.png")
@@ -1021,13 +1021,13 @@ def test_generate_data_with_arabic_text(self):
10211021
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
10221022
)
10231023

1024-
self.assertTrue(
1024+
self.assertLess(
10251025
diff(
10261026
"tests/out/21_اختبار اختبار اختبار.png",
10271027
"tests/expected_results/21_اختبار اختبار اختبار.png",
10281028
delete_diff_file=True,
1029-
)
1030-
< 0.01
1029+
),
1030+
0.05
10311031
)
10321032

10331033
os.remove("tests/out/21_اختبار اختبار اختبار.png")
@@ -1062,13 +1062,13 @@ def test_generate_data_with_hindi_text(self):
10621062
os.path.join(os.path.split(os.path.realpath(__file__))[0], "trdg/images"),
10631063
)
10641064

1065-
self.assertTrue(
1065+
self.assertLess(
10661066
diff(
10671067
"tests/out/22_परीक्षा परीक्षा परीक्षा.png",
10681068
"tests/expected_results/22_परीक्षा परीक्षा परीक्षा.png",
10691069
delete_diff_file=True,
1070-
)
1071-
< 0.01
1070+
),
1071+
0.17
10721072
)
10731073

10741074
os.remove("tests/out/22_परीक्षा परीक्षा परीक्षा.png")

0 commit comments

Comments
 (0)