Merge pull request #2 from knikolaevskii/Rov's-branch

Rov's branch

Author: knikolaevskii

src/prompt_toolkit/layout/dimension.py

@@ -182,20 +182,39 @@ def max_layout_dimensions(dimensions: list[Dimension]) -> Dimension:
     Callable[[], Any],
 ]
 
+from typing import Any, Callable, Union
+
+to_dimension_coverage = {
+    "to_dimension_1": False,  
+    "to_dimension_2": False, 
+    "to_dimension_3": False,  
+    "to_dimension_4": False,  
+    "to_dimension_5": False   
+}
 
 def to_dimension(value: AnyDimension) -> Dimension:
     """
     Turn the given object into a `Dimension` object.
     """
     if value is None:
+        to_dimension_coverage["to_dimension_1"] = True
+        print("Branch 1 was hit")
         return Dimension()
     if isinstance(value, int):
+        to_dimension_coverage["to_dimension_2"] = True
+        print("Branch 2 was hit")
         return Dimension.exact(value)
     if isinstance(value, Dimension):
+        to_dimension_coverage["to_dimension_3"] = True
+        print("Branch 3 was hit")
         return value
     if callable(value):
+        to_dimension_coverage["to_dimension_4"] = True
+        print("Branch 4 was hit")
         return to_dimension(value())
 
+    to_dimension_coverage["to_dimension_5"] = True
+    print("Branch 5 was hit")
     raise ValueError("Not an integer or Dimension object.")
 
 

src/prompt_toolkit/layout/processors.py

@@ -943,16 +943,25 @@ def apply_transformation(self, ti: TransformationInput) -> Transformation:
         return processor.apply_transformation(ti)
 
 
+merge_processors_coverage = {
+    "merge_processors_1": False,  # Branch for empty list of processors
+    "merge_processors_2": False,  # Branch for a single processor
+    "merge_processors_3": False   # Branch for multiple processors
+}
+
 def merge_processors(processors: list[Processor]) -> Processor:
     """
     Merge multiple `Processor` objects into one.
     """
     if len(processors) == 0:
+        merge_processors_coverage["merge_processors_1"] = True
         return DummyProcessor()
 
     if len(processors) == 1:
+        merge_processors_coverage["merge_processors_2"] = True
         return processors[0]  # Nothing to merge.
 
+    merge_processors_coverage["merge_processors_3"] = True
     return _MergedProcessor(processors)
 
 

tests/test_merge_processors.py

@@ -0,0 +1,28 @@
+from prompt_toolkit.layout.processors import merge_processors, merge_processors_coverage
+from prompt_toolkit.layout.processors import Processor, DummyProcessor, _MergedProcessor
+
+def print_coverage():
+    print("\nCoverage report:")
+    hit_branches = sum(merge_processors_coverage.values())
+    total_branches = len(merge_processors_coverage)
+    coverage_percentage = (hit_branches / total_branches) * 100
+    for branch, hit in merge_processors_coverage.items():
+        print(f"{branch} was {'hit' if hit else 'not hit'}")
+    print(f"Coverage: {hit_branches}/{total_branches} branches hit ({coverage_percentage:.2f}%)\n")
+
+class MockProcessor(Processor):
+    def apply_transformation(self, *args, **kwargs):
+        pass
+
+test_cases = [
+    ([], "Empty list of processors (should hit Branch 1)"),
+    ([MockProcessor()], "Single processor (should hit Branch 2)"),
+    ([MockProcessor(), MockProcessor()], "Multiple processors (should hit Branch 3)")
+]
+
+for processors, description in test_cases:
+    print(f"\nTesting case: {description}")
+    result = merge_processors(processors)
+    print(f"Result: {result}")
+
+print_coverage()

tests/test_to_dimension.py

@@ -0,0 +1,29 @@
+from prompt_toolkit.layout.dimension import Dimension, to_dimension, to_dimension_coverage
+
+def print_coverage():
+    print("\nCoverage report:")
+    hit_branches = sum(to_dimension_coverage.values())
+    total_branches = len(to_dimension_coverage)
+    coverage_percentage = (hit_branches / total_branches) * 100
+    for branch, hit in to_dimension_coverage.items():
+        print(f"{branch} was {'hit' if hit else 'not hit'}")
+    print(f"Coverage: {hit_branches}/{total_branches} branches hit ({coverage_percentage:.2f}%)\n")
+
+
+test_cases = [
+    (None, "None value (should hit Branch 1)"),
+    (69, "Integer value (should hit Branch 2)"),
+    (Dimension(), "Dimension instance (should hit Branch 3)"),
+    (lambda: 42, "Callable returning an integer (should hit Branch 4)"),
+    ("Unsupported type", "Unsupported type (should hit Branch 5)")
+]
+
+for value, description in test_cases:
+    print(f"\nTesting case: {description}")
+    try:
+        result = to_dimension(value)
+        print(f"Result: {result}")
+    except ValueError as e:
+        print(f"Exception: {e}")
+
+print_coverage()