stouputils 1.2.12__tar.gz → 1.2.14__tar.gz

{stouputils-1.2.12 → stouputils-1.2.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stouputils
-Version: 1.2.12
+Version: 1.2.14
 Summary: Stouputils is a collection of utility modules designed to simplify and enhance the development process. It includes a range of tools for tasks such as execution of doctests, display utilities, decorators, as well as context managers, and many more.
 Project-URL: Homepage, https://github.com/Stoupy51/stouputils
 Project-URL: Issues, https://github.com/Stoupy51/stouputils/issues

{stouputils-1.2.12 → stouputils-1.2.14}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "stouputils"
-version = "1.2.12"
+version = "1.2.14"
 description = "Stouputils is a collection of utility modules designed to simplify and enhance the development process. It includes a range of tools for tasks such as execution of doctests, display utilities, decorators, as well as context managers, and many more."
 readme = "README.md"
 requires-python = ">=3.10"

{stouputils-1.2.12 → stouputils-1.2.14}/stouputils/__init__.py

@@ -25,6 +25,7 @@ from .parallel import *
 from .all_doctests import *
 from .collections import *
 from .backup import *
+from .image import *
 
 # Folders
 from .continuous_delivery import *

{stouputils-1.2.12 → stouputils-1.2.14}/stouputils/all_doctests.py

@@ -23,7 +23,7 @@ def test_module_with_progress(module: ModuleType, separator: str) -> TestResults
 	return internal()
 
 # Main program
-def launch_tests(root_dir: str, importing_errors: LogLevels = LogLevels.WARNING_TRACEBACK, strict: bool = True) -> None:
+def launch_tests(root_dir: str, importing_errors: LogLevels = LogLevels.WARNING_TRACEBACK, strict: bool = True) -> int:
 	""" Main function to launch tests for all modules in the given directory.
 
 	Args:
@@ -31,6 +31,9 @@ def launch_tests(root_dir: str, importing_errors: LogLevels = LogLevels.WARNING_
 		importing_errors		(LogLevels):	Log level for the errors when importing modules
 		strict					(bool):			Modify the force_raise_exception variable to True in the decorators module
 	
+	Returns:
+		int: The number of failed tests
+	
 	Examples:
 		>>> launch_tests("unknown_dir")
 		Traceback (most recent call last):
@@ -39,7 +42,8 @@ def launch_tests(root_dir: str, importing_errors: LogLevels = LogLevels.WARNING_
 
 	.. code-block:: python
 
-		> launch_tests("/path/to/source")
+		> if launch_tests("/path/to/source") > 0:
+			sys.exit(1)
 		[PROGRESS HH:MM:SS] Importing module 'module1'	took 0.001s
 		[PROGRESS HH:MM:SS] Importing module 'module2'	took 0.002s
 		[PROGRESS HH:MM:SS] Importing module 'module3'	took 0.003s
@@ -89,10 +93,15 @@ def launch_tests(root_dir: str, importing_errors: LogLevels = LogLevels.WARNING_
 	results: list[TestResults] = [test_module_with_progress(module, separator) for module, separator in zip(modules, separators)]
 
 	# Display any error lines for each module at the end of the script
+	nb_failed_tests: int = 0
 	for module, result in zip(modules, results):
 		if result.failed > 0:
 			error(f"Errors in module {module.__name__}", exit=False)
+			nb_failed_tests += result.failed
 
 	# Reset force_raise_exception back
 	decorators.force_raise_exception = strict
 
+	# Return the number of failed tests
+	return nb_failed_tests
+

stouputils-1.2.14/stouputils/image.py

@@ -0,0 +1,70 @@
+
+# Imports
+from PIL import Image
+from typing import Callable
+from numpy.typing import NDArray
+import numpy as np
+
+
+# Functions
+def image_resize(
+	image: Image.Image | NDArray[np.uint8],
+	max_result_size: int,
+	resampling: Image.Resampling = Image.Resampling.LANCZOS,
+	min_or_max: Callable[[int, int], int] = max,
+	return_type: type[Image.Image | NDArray[np.uint8]] = Image.Image,
+) -> Image.Image | NDArray[np.uint8]:
+	""" Resize an image while preserving its aspect ratio.
+	Scales the image so that its largest dimension equals max_result_size.
+	
+	Args:
+		image             (Image.Image | np.ndarray): The image to resize.
+		max_result_size   (int):                      Maximum size for the largest dimension.
+		resampling        (Image.Resampling):         PIL resampling filter to use.
+		min_or_max        (Callable):                 Function to use to get the minimum or maximum of the two ratios.
+		return_type       (type):                     Type of the return value (Image.Image or np.ndarray).
+	Returns:
+		Image.Image: The resized image with preserved aspect ratio.
+	Examples:
+		>>> # Test with (height x width x channels) numpy array
+		>>> import numpy as np
+		>>> array: np.ndarray = np.random.randint(0, 255, (100, 50, 3), dtype=np.uint8)
+		>>> image_resize(array, 100).size
+		(50, 100)
+		>>> image_resize(array, 100, min_or_max=max).size
+		(50, 100)
+		>>> image_resize(array, 100, min_or_max=min).size
+		(100, 200)
+
+		>>> # Test with PIL Image
+		>>> from PIL import Image
+		>>> pil_image: Image.Image = Image.new('RGB', (200, 100))
+		>>> image_resize(pil_image, 50).size
+		(50, 25)
+		>>> # Test with different return types
+		>>> resized_array = image_resize(array, 50, return_type=np.ndarray)
+		>>> isinstance(resized_array, np.ndarray)
+		True
+		>>> resized_array.shape
+		(50, 25, 3)
+		>>> # Test with different resampling methods
+		>>> image_resize(pil_image, 50, resampling=Image.Resampling.NEAREST).size
+		(50, 25)
+	"""
+	if isinstance(image, np.ndarray):
+		image = Image.fromarray(image)	# type: ignore
+
+	width: int = image.size[0]
+	height: int = image.size[1]
+	max_dimension: int = min_or_max(width, height)
+	scale: float = max_result_size / max_dimension
+	
+	new_width: int = int(width * scale)
+	new_height: int = int(height * scale)
+	new_image: Image.Image = image.resize((new_width, new_height), resampling)
+	
+	if return_type == np.ndarray:
+		return np.array(new_image)
+	else:
+		return new_image
+