repairing tests (7d47b152) · Commits · Filip Kučerák / kdtesting

kdtesting.py

+6 −6

Original line number	Diff line number	Diff line
		@@ -62,17 +62,17 @@ def upper_case_let_gen() -> Callable[[], str]:
		return generator


		def str_gen(generator: Callable[[], str], min_size: int, max_size: int) \
		def str_gen(char_generator: Callable[[], str], min_size: int, max_size: int) \
		-> Callable[[], str]:
		"""
		Returns:
		Callable[[], str]: value generator that returns string with size in
		range [min_size, max_size)
		"""
		char_list_generator = listof_gen(generator, min_size, max_size)
		char_list_generator = listof_gen(char_generator, min_size, max_size)

		def generator() -> str:
		return "".join(char_list_generator)
		return "".join(char_list_generator())
		return generator

		####################
		@@ -151,7 +151,7 @@ def eqchoices_gen(choices: List[Callable[[], T]]) \
		Callable[[], T]: value generator that returns value provided by one of
		the generators, the generators have equal chance of being chosen
		"""
		def generator() -> T:
		def generator() -> Tuple[int, T]:
		index = random.randrange(0, len(choices))
		return (index, choices[index]())
		return generator
		@@ -251,7 +251,7 @@ class Unshrinkable(Arbitrary[T]):
		shrinks to nothing.
		"""

		def __init__(self, generator: Arbitrary[T]) -> None:
		def __init__(self, generator: Callable[[], T]) -> None:
		super().__init__(generator)

		def shrink(self, value: T) -> Generator[T, None, None]:
		@@ -385,7 +385,7 @@ class NumberA(Arbitrary[int]):
		yield abs(value - 1) * (abs(value) // value)


		class StringA(ListA[List[str]]):
		class StringA(ListA[str]):
		"""
		StringA returns random string of alpha characters with length in range
		[min_value, max_value)