PythonSnips/pystrings/similar_string_groups/test_similar_string_groups.py at 69e29987af7de878218bfe00d30ee241e5f03616 · BrianLusina/PythonSnips · 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
import unittest
from . import num_similar_groups, num_similar_groups_2


class SimilarStringGroupsTestCase(unittest.TestCase):
    def test_1(self):
        strs = ["jhki", "kijh", "jkhi", "kihj", "ijhk"]
        expected = 3
        actual = num_similar_groups(strs)
        self.assertEqual(expected, actual)

    def test_2(self):
        strs = ["abc", "acb", "bac", "bca", "cab", "cba"]
        expected = 1
        actual = num_similar_groups(strs)
        self.assertEqual(expected, actual)

    def test_3(self):
        strs = ["abcd", "abdc", "acbd", "bdca"]
        expected = 2
        actual = num_similar_groups(strs)
        self.assertEqual(expected, actual)

    def test_4(self):
        strs = ["fgtdvepeqcfajhlzkwlpuhrwfcueqfbs","fgcdvppeqcfajhlzkwluehrwftuefqbs","fgtdvepeqcfajhlzkwlpuhrwfcuefqbs",
                "fgcdvepeqcfajhlzkwluphrwftuefqbs","fgldvepeqcfajhlzkwcuphrwftuefqbs","fgtdvefeqcpajhlzkwlpuhrwfcuefqbs"]
        expected = 2
        actual = num_similar_groups(strs)
        self.assertEqual(expected, actual)


class SimilarStringGroups2TestCase(unittest.TestCase):
    def test_1(self):
        strs = ["jhki", "kijh", "jkhi", "kihj", "ijhk"]
        expected = 3
        actual = num_similar_groups_2(strs)
        self.assertEqual(expected, actual)

    def test_2(self):
        strs = ["abc", "acb", "bac", "bca", "cab", "cba"]
        expected = 1
        actual = num_similar_groups_2(strs)
        self.assertEqual(expected, actual)

    def test_3(self):
        strs = ["abcd", "abdc", "acbd", "bdca"]
        expected = 2
        actual = num_similar_groups_2(strs)
        self.assertEqual(expected, actual)

    def test_4(self):
        strs = ["fgtdvepeqcfajhlzkwlpuhrwfcueqfbs","fgcdvppeqcfajhlzkwluehrwftuefqbs","fgtdvepeqcfajhlzkwlpuhrwfcuefqbs",
                "fgcdvepeqcfajhlzkwluphrwftuefqbs","fgldvepeqcfajhlzkwcuphrwftuefqbs","fgtdvefeqcpajhlzkwlpuhrwfcuefqbs"]
        expected = 2
        actual = num_similar_groups_2(strs)
        self.assertEqual(expected, actual)


if __name__ == '__main__':
    unittest.main()