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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
import re
import unittest
import random
from rstr import Rstr, SameCharacterError
def assert_matches(pattern: str, value: str) -> None:
errmsg = f'{value} does not match {pattern}'
assert re.match(pattern, value), errmsg
class TestRstr(unittest.TestCase):
def setUp(self) -> None:
self.rs = Rstr()
def test_specific_length(self) -> None:
assert_matches('^A{5}$', self.rs.rstr('A', 5))
def test_length_range(self) -> None:
assert_matches('^A{11,20}$', self.rs.rstr('A', 11, 20))
def test_end_range_no_start_range(self) -> None:
assert_matches('^A{1,20}$', self.rs.rstr('A', end_range=20))
def test_custom_alphabet(self) -> None:
assert_matches('^A{1,10}$', self.rs.rstr('AA'))
def test_alphabet_as_list(self) -> None:
assert_matches('^A{1,10}$', self.rs.rstr(['A', 'A']))
def test_include(self) -> None:
assert_matches('^[ABC]*@[ABC]*$', self.rs.rstr('ABC', include='@'))
def test_include_specific_length(self) -> None:
'''
Verify including characters doesn't make the string longer than intended.
'''
assert_matches('^[ABC@]{5}$', self.rs.rstr('ABC', 5, include='@'))
def test_exclude(self) -> None:
for _ in range(0, 100):
assert 'C' not in self.rs.rstr('ABC', exclude='C')
def test_include_as_list(self) -> None:
assert_matches('^[ABC]*@[ABC]*$', self.rs.rstr('ABC', include=['@']))
def test_exclude_as_list(self) -> None:
for _ in range(0, 100):
assert 'C' not in self.rs.rstr('ABC', exclude=['C'])
def test_raise_exception_if_include_and_exclude_parameters_contain_same_character(self) -> None:
with self.assertRaisesRegex(SameCharacterError, r"include and exclude parameters contain same character \(B\)"):
self.rs.rstr('A', include='B', exclude='B')
self.rs.rstr('A', include=['B'], exclude=['B'])
with self.assertRaisesRegex(SameCharacterError, r"include and exclude parameters contain same characters \(., .\)"):
self.rs.rstr('A', include='BC', exclude='BC')
class TestSystemRandom(TestRstr):
def setUp(self) -> None:
self.rs = Rstr(random.SystemRandom())
class TestDigits(unittest.TestCase):
def setUp(self) -> None:
self.rs = Rstr()
def test_all_digits(self) -> None:
assert_matches(r'^\d{1,10}$', self.rs.digits())
def test_digits_include(self) -> None:
assert_matches(r'^\d*@\d*$', self.rs.digits(include='@'))
def test_digits_exclude(self) -> None:
for _ in range(0, 100):
assert '5' not in self.rs.digits(exclude='5')
class TestNondigits(unittest.TestCase):
def setUp(self) -> None:
self.rs = Rstr()
def test_nondigits(self) -> None:
assert_matches(r'^\D{1,10}$', self.rs.nondigits())
def test_nondigits_include(self) -> None:
assert_matches(r'^\D*@\D*$', self.rs.nondigits(include='@'))
def test_nondigits_exclude(self) -> None:
for _ in range(0, 100):
assert 'A' not in self.rs.nondigits(exclude='A')
class TestLetters(unittest.TestCase):
def setUp(self) -> None:
self.rs = Rstr()
def test_letters(self) -> None:
assert_matches(r'^[a-zA-Z]{1,10}$', self.rs.letters())
def test_letters_include(self) -> None:
assert_matches(r'^[a-zA-Z]*@[a-zA-Z]*$', self.rs.letters(include='@'))
def test_letters_exclude(self) -> None:
for _ in range(0, 100):
assert 'A' not in self.rs.letters(exclude='A')
class TestUnambiguous(unittest.TestCase):
def setUp(self) -> None:
self.rs = Rstr()
def test_unambiguous(self) -> None:
assert_matches('^[a-km-zA-HJ-NP-Z2-9]{1,10}$', self.rs.unambiguous())
def test_unambiguous_include(self) -> None:
assert_matches('^[a-km-zA-HJ-NP-Z2-9@]{1,10}$', self.rs.unambiguous(include='@'))
def test_unambiguous_exclude(self) -> None:
for _ in range(0, 100):
assert 'A' not in self.rs.unambiguous(exclude='A')
class TestCustomAlphabets(unittest.TestCase):
def test_alphabet_at_instantiation(self) -> None:
rs = Rstr(vowels='AEIOU')
assert_matches('^[AEIOU]{1,10}$', rs.vowels())
def test_add_alphabet(self) -> None:
rs = Rstr()
rs.add_alphabet('evens', '02468')
assert_matches('^[02468]{1,10}$', rs.evens())
def main() -> None:
unittest.main()
if __name__ == '__main__':
main()
|
