|
1 | 1 | import unittest
|
2 | 2 |
|
3 |
| -from betterversion.better import version_windows, BetterVersion |
| 3 | +from betterversion.better import version_windows, VersionIdentifier |
4 | 4 |
|
5 |
| -class TestSequenceFunctions(unittest.TestCase): |
6 | 5 |
|
| 6 | +class TestSequenceFunctions(unittest.TestCase): |
7 | 7 | def setUp(self):
|
8 | 8 | self.versions = [
|
9 |
| - BetterVersion("0.1.0"), |
10 |
| - BetterVersion("0.2.0"), |
11 |
| - BetterVersion("0.2.1"), |
12 |
| - BetterVersion("0.3.0"), |
13 |
| - BetterVersion("0.3.1"), |
14 |
| - BetterVersion("1.1.0"), |
15 |
| - BetterVersion("1.2.0"), |
16 |
| - BetterVersion("1.3.0"), |
17 |
| - BetterVersion("2.1.0"), |
18 |
| - BetterVersion("2.2.0"), |
19 |
| - BetterVersion("2.3.0"), |
20 |
| - BetterVersion("2.3.1"), |
21 |
| - BetterVersion("2.3.2"), |
22 |
| - BetterVersion("2.3.3"), |
| 9 | + VersionIdentifier("0.1.0"), |
| 10 | + VersionIdentifier("0.2.0"), |
| 11 | + VersionIdentifier("0.2.1"), |
| 12 | + VersionIdentifier("0.3.0"), |
| 13 | + VersionIdentifier("0.3.1"), |
| 14 | + VersionIdentifier("1.1.0"), |
| 15 | + VersionIdentifier("1.2.0"), |
| 16 | + VersionIdentifier("1.3.0"), |
| 17 | + VersionIdentifier("2.1.0"), |
| 18 | + VersionIdentifier("2.2.0"), |
| 19 | + VersionIdentifier("2.3.0"), |
| 20 | + VersionIdentifier("2.3.1"), |
| 21 | + VersionIdentifier("2.3.2"), |
| 22 | + VersionIdentifier("2.3.3"), |
23 | 23 | ]
|
24 | 24 |
|
25 | 25 | def test_major(self):
|
@@ -65,40 +65,59 @@ def test_point(self):
|
65 | 65 | self.assertEqual(len(point_versions[2][3]), 4)
|
66 | 66 |
|
67 | 67 | def test_sort(self):
|
68 |
| - final_versions = version_windows(self.versions, major=2, minor=2, point=1) |
69 |
| - self.assertTrue(final_versions[1][2][0] == BetterVersion('1.2.0')) |
70 |
| - self.assertTrue(final_versions[1][3][0] == BetterVersion('1.3.0')) |
71 |
| - self.assertTrue(final_versions[2][3][0] == BetterVersion('2.3.3')) |
72 |
| - self.assertTrue(final_versions[2][3][0] != BetterVersion('2.3.0')) |
73 |
| - |
74 |
| - final_versions = version_windows(self.versions, major=1, minor=2, point=2) |
| 68 | + final_versions = version_windows(self.versions, |
| 69 | + major=2, minor=2, point=1) |
| 70 | + self.assertTrue(final_versions[1][2][0] == VersionIdentifier('1.2.0')) |
| 71 | + self.assertTrue(final_versions[1][3][0] == VersionIdentifier('1.3.0')) |
| 72 | + self.assertTrue(final_versions[2][3][0] == VersionIdentifier('2.3.3')) |
| 73 | + self.assertTrue(final_versions[2][3][0] != VersionIdentifier('2.3.0')) |
| 74 | + |
| 75 | + final_versions = version_windows(self.versions, |
| 76 | + major=1, minor=2, point=2) |
75 | 77 | # 1 major version
|
76 | 78 | self.assertEqual(final_versions[1], {})
|
77 | 79 | # 2 minor versions
|
78 | 80 | self.assertEqual(len(final_versions[2]), 2)
|
79 | 81 | # 2 point versions
|
80 | 82 | self.assertEqual(len(final_versions[2][3]), 2)
|
81 |
| - final_versions = version_windows(self.versions, major=1, minor=2, point=3) |
| 83 | + final_versions = version_windows(self.versions, |
| 84 | + major=1, minor=2, point=3) |
82 | 85 | # 2 point versions
|
83 | 86 | self.assertEqual(len(final_versions[2][3]), 3)
|
84 | 87 |
|
85 | 88 | def test_flat(self):
|
86 |
| - final_versions = version_windows(self.versions, major=2, minor=2, point=1, flat=True) |
| 89 | + final_versions = version_windows(self.versions, |
| 90 | + major=2, minor=2, point=1, flat=True) |
87 | 91 | self.assertEqual(len(final_versions), 4)
|
88 | 92 | self.assertEqual(
|
89 | 93 | final_versions,
|
90 |
| - [BetterVersion('1.2.0'), BetterVersion('1.3.0'), BetterVersion('2.2.0'), BetterVersion('2.3.3')] |
| 94 | + [ |
| 95 | + VersionIdentifier('1.2.0'), |
| 96 | + VersionIdentifier('1.3.0'), |
| 97 | + VersionIdentifier('2.2.0'), |
| 98 | + VersionIdentifier('2.3.3') |
| 99 | + ] |
91 | 100 | )
|
92 | 101 |
|
93 |
| - final_versions = version_windows(self.versions, major=3, minor=2, point=1, flat=True) |
| 102 | + final_versions = version_windows(self.versions, |
| 103 | + major=3, minor=2, point=1, flat=True) |
94 | 104 | self.assertEqual(len(final_versions), 6)
|
95 | 105 | self.assertEqual(
|
96 | 106 | final_versions,
|
97 |
| - [BetterVersion('0.2.1'), BetterVersion('0.3.1'), BetterVersion('1.2.0'), BetterVersion('1.3.0'), BetterVersion('2.2.0'), BetterVersion('2.3.3')] |
| 107 | + [ |
| 108 | + VersionIdentifier('0.2.1'), |
| 109 | + VersionIdentifier('0.3.1'), |
| 110 | + VersionIdentifier('1.2.0'), |
| 111 | + VersionIdentifier('1.3.0'), |
| 112 | + VersionIdentifier('2.2.0'), |
| 113 | + VersionIdentifier('2.3.3') |
| 114 | + ] |
98 | 115 | )
|
99 | 116 |
|
100 |
| - final_versions = version_windows(self.versions, major=3, minor=2, point=2, flat=True) |
| 117 | + final_versions = version_windows(self.versions, |
| 118 | + major=3, minor=2, point=2, flat=True) |
101 | 119 | self.assertEqual(len(final_versions), 9)
|
102 | 120 |
|
| 121 | + |
103 | 122 | if __name__ == '__main__':
|
104 | 123 | unittest.main()
|
0 commit comments