assignment-text-processing/test_text_processing.py at main · TEAMLAB-Lecture/assignment-text-processing · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
# -*- coding: utf8 -*-

import unittest
import random
import test_processing as tp


class TestTextProcessing(unittest.TestCase):
    def test_normalize(self):
        test_str = "This is an example."
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "this is an example.")

        test_str = "   EXTRA   SPACE   "
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "extra space")

        test_str = "THIS IS ALL CAPS!!"
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "this is all caps!!")

        test_str = "                   "
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "")

        test_str = "this is all lower space..."
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "this is all lower space...")

        test_str = "  H  e  L    l   O   !"
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "h e l l o !")

        test_str = ""
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "")

        test_str = "........"
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "........")

        test_str = "EX  A M P     LE"
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "ex a m p le")

        test_str = "Test Text Normalization"
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "test text normalization")

        test_str = "AbCd EfGh IjKl MnOp"
        pred = tp.normalize(test_str)
        self.assertEqual(pred, "abcd efgh ijkl mnop")

    def test_no_vowels(self):
        test_str = "This is an example."
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "Ths s n xmpl.")

        test_str = "We love Python!"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "W lv Pythn!")

        test_str = "AEIOU"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "")

        test_str = "aeiou"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "")

        test_str = "QWERTY"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "QWRTY")

        test_str = "qwerty"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "qwrty")

        test_str = "AI for ALL!"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, " fr LL!")

        test_str = "Are there any vowels?"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "r thr ny vwls?")

        test_str = ""
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "")

        test_str = "abcdefghijklmnopqrstuvwxyz"
        pred = tp.no_vowels(test_str)
        self.assertEqual(pred, "bcdfghjklmnpqrstvwxyz")