class OneCharStringStrategy(SearchStrategy):
"""A strategy which generates single character strings of text type."""
specifier = text_type
zero_point = ord('0')
def __init__(self,
whitelist_categories=None,
blacklist_categories=None,
blacklist_characters=None,
min_codepoint=None,
max_codepoint=None):
intervals = charmap.query(
include_categories=whitelist_categories,
exclude_categories=blacklist_categories,
min_codepoint=min_codepoint,
max_codepoint=max_codepoint,
)
if not intervals:
raise InvalidArgument('No valid characters in set')
self.intervals = IntervalSet(intervals)
if blacklist_characters:
self.blacklist_characters = set(b for b in blacklist_characters
if ord(b) in self.intervals)
if len(self.blacklist_characters) == len(self.intervals):
raise InvalidArgument('No valid characters in set')
else:
self.blacklist_characters = set()
self.zero_point = self.intervals.index_above(ord('0'))
self.special = []
if '\n' not in self.blacklist_characters:
n = ord('\n')
try:
self.special.append(self.intervals.index(n))
except ValueError:
pass
def do_draw(self, data):
denom = math.log1p(-1 / 127)
def d(random):
if self.special and random.randint(0, 10) == 0:
return random.choice(self.special)
if len(self.intervals) <= 256 or random.randint(0, 1):
i = random.randint(0, len(self.intervals.offsets) - 1)
u, v = self.intervals.intervals[i]
return self.intervals.offsets[i] + random.randint(0, v - u + 1)
else:
return min(
len(self.intervals) - 1,
int(math.log(random.random()) / denom))
while True:
i = integer_range(data,
0,
len(self.intervals) - 1,
center=self.zero_point,
distribution=d)
c = hunichr(self.intervals[i])
if c not in self.blacklist_characters:
return c
class OneCharStringStrategy(SearchStrategy):
"""A strategy which generates single character strings of text type."""
specifier = text_type
zero_point = ord('0')
def __init__(self,
whitelist_categories=None,
blacklist_categories=None,
blacklist_characters=None,
min_codepoint=None,
max_codepoint=None):
intervals = charmap.query(
include_categories=whitelist_categories,
exclude_categories=blacklist_categories,
min_codepoint=min_codepoint,
max_codepoint=max_codepoint,
)
if not intervals:
raise InvalidArgument(
'No valid characters in set'
)
self.intervals = IntervalSet(intervals)
if blacklist_characters:
self.blacklist_characters = set(
b for b in blacklist_characters if ord(b) in self.intervals
)
if len(self.blacklist_characters) == len(self.intervals):
raise InvalidArgument(
'No valid characters in set'
)
else:
self.blacklist_characters = set()
self.zero_point = self.intervals.index_above(ord('0'))
self.special = []
if '\n' not in self.blacklist_characters:
n = ord('\n')
try:
self.special.append(self.intervals.index(n))
except ValueError:
pass
def do_draw(self, data):
denom = math.log1p(-1 / 127)
def d(random):
if self.special and random.randint(0, 10) == 0:
return random.choice(self.special)
if len(self.intervals) <= 256 or random.randint(0, 1):
i = random.randint(0, len(self.intervals.offsets) - 1)
u, v = self.intervals.intervals[i]
return self.intervals.offsets[i] + random.randint(0, v - u + 1)
else:
return min(
len(self.intervals) - 1,
int(math.log(random.random()) / denom))
while True:
i = integer_range(
data, 0, len(self.intervals) - 1,
center=self.zero_point, distribution=d
)
c = hunichr(self.intervals[i])
if c not in self.blacklist_characters:
return c
