/usr/lib/python3/dist-packages/rstr/tests/test_rstr.py is in python3-rstr 2.2.6-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 | import re
import unittest
import sys
import random
from rstr.rstr_base import Rstr
if sys.version_info[0] >= 3:
unichr = chr
xrange = range
class TestRstr(unittest.TestCase):
def setUp(self):
self.rs = Rstr()
def test_specific_length(self):
assert re.match('^A{5}$', self.rs.rstr('A', 5))
def test_length_range(self):
assert re.match('^A{11,20}$', self.rs.rstr('A', 11, 20))
def test_custom_alphabet(self):
assert re.match('^A{1,10}$', self.rs.rstr('AA'))
def test_alphabet_as_list(self):
assert re.match('^A{1,10}$', self.rs.rstr(['A', 'A']))
def test_include(self):
assert re.match('^[ABC]*@[ABC]*$', self.rs.rstr('ABC', include='@'))
def test_exclude(self):
for _ in xrange(0, 100):
assert 'C' not in self.rs.rstr('ABC', exclude='C')
def test_include_as_list(self):
assert re.match('^[ABC]*@[ABC]*$', self.rs.rstr('ABC', include=["@"]))
def test_exclude_as_list(self):
for _ in xrange(0, 100):
assert 'C' not in self.rs.rstr('ABC', exclude=['C'])
class TestSystemRandom(TestRstr):
def setUp(self):
self.rs = Rstr(random.SystemRandom())
class TestDigits(unittest.TestCase):
def setUp(self):
self.rs = Rstr()
def test_all_digits(self):
assert re.match('^\d{1,10}$', self.rs.digits())
def test_digits_include(self):
assert re.match('^\d*@\d*$', self.rs.digits(include='@'))
def test_digits_exclude(self):
for _ in xrange(0, 100):
assert '5' not in self.rs.digits(exclude='5')
class TestNondigits(unittest.TestCase):
def setUp(self):
self.rs = Rstr()
def test_nondigits(self):
assert re.match('^\D{1,10}$', self.rs.nondigits())
def test_nondigits_include(self):
assert re.match('^\D*@\D*$', self.rs.nondigits(include='@'))
def test_nondigits_exclude(self):
for _ in xrange(0, 100):
assert 'A' not in self.rs.nondigits(exclude='A')
class TestLetters(unittest.TestCase):
def setUp(self):
self.rs = Rstr()
def test_letters(self):
assert re.match('^[a-zA-Z]{1,10}$', self.rs.letters())
def test_letters_include(self):
assert re.match('^[a-zA-Z]*@[a-zA-Z]*$', self.rs.letters(include='@'))
def test_letters_exclude(self):
for _ in xrange(0, 100):
assert 'A' not in self.rs.letters(exclude='A')
class TestCustomAlphabets(unittest.TestCase):
def test_alphabet_at_instantiation(self):
rs = Rstr(vowels='AEIOU')
assert re.match('^[AEIOU]{1,10}$', rs.vowels())
def test_add_alphabet(self):
rs = Rstr()
rs.add_alphabet('evens', '02468')
assert re.match('^[02468]{1,10}$', rs.evens())
def main():
unittest.main()
if __name__ == '__main__':
main()
|