test_triangle_coverage.py

from enum import Enum
from typing import Set
from coverage_report import (
    print_test_execution,
    print_coverage_summary,
    print_branch_coverage_table,
    print_latex_table,
    run_tests,
)


class TriangleType(Enum):
    NOT_A_TRIANGLE = "NOT_A_TRIANGLE"
    SCALENE = "SCALENE"
    EQUILATERAL = "EQUILATERAL"
    ISOSCELES = "ISOSCELES"


def classify(num1: int, num2: int, num3: int, covered_stmts: Set[int], covered_branches: Set[str]) -> TriangleType:
    type_result = None
    
    # Branch 1: num1 > num2?
    if num1 > num2:
        covered_branches.add("1T")
        covered_stmts.add(1)  # temp = num1
        temp = num1
        covered_stmts.add(2)  # num1 = num2
        num1 = num2
        covered_stmts.add(3)  # num2 = temp
        num2 = temp
    else:
        covered_branches.add("1F")
    
    # Branch 2: num1 > num3?
    if num1 > num3:
        covered_branches.add("2T")
        covered_stmts.add(4)  # temp = num1
        temp = num1
        covered_stmts.add(5)  # num1 = num3
        num1 = num3
        covered_stmts.add(6)  # num3 = temp
        num3 = temp
    else:
        covered_branches.add("2F")
    
    # Branch 3: num2 > num3?
    if num2 > num3:
        covered_branches.add("3T")
        covered_stmts.add(7)  # temp = num2
        temp = num2
        covered_stmts.add(8)  # num2 = num3
        num2 = num3
        covered_stmts.add(9)  # num3 = temp
        num3 = temp
    else:
        covered_branches.add("3F")
    
    # Branch 4: num1 + num2 <= num3?
    if num1 + num2 <= num3:
        covered_branches.add("4T")
        covered_stmts.add(10)  # type_result = TriangleType.NOT_A_TRIANGLE
        type_result = TriangleType.NOT_A_TRIANGLE
    else:
        covered_branches.add("4F")
        covered_stmts.add(11)  # type_result = TriangleType.SCALENE
        type_result = TriangleType.SCALENE
        # Branch 5: num1 == num2?
        if num1 == num2:
            covered_branches.add("5T")
            # Branch 6: num2 == num3?
            if num2 == num3:
                covered_branches.add("6T")
                covered_stmts.add(12)  # type_result = TriangleType.EQUILATERAL
                type_result = TriangleType.EQUILATERAL
            else:
                covered_branches.add("6F")
        else:
            covered_branches.add("5F")
            # Branch 7: num1 == num2?
            if num1 == num2:
                covered_branches.add("7T")
                covered_stmts.add(13)  # type_result = TriangleType.ISOSCELES
                type_result = TriangleType.ISOSCELES
            else:
                covered_branches.add("7F")
            # Branch 8: num2 == num3?
            if num2 == num3:
                covered_branches.add("8T")
                covered_stmts.add(14)  # type_result = TriangleType.ISOSCELES
                type_result = TriangleType.ISOSCELES
            else:
                covered_branches.add("8F")
    
    covered_stmts.add(15)  # return type_result
    return type_result


def main():
    test_cases = {
        "(2,3,3)": (2, 3, 3),
        "(3,2,1)": (3, 2, 1),
        "(3,3,3)": (3, 3, 3),
        "(2,2,3)": (2, 2, 3),
        "(4,5,6)": (4, 5, 6),
    }
    
    test_results, total_covered_stmts, total_covered_branches = run_tests(test_cases, classify)
    
    print_test_execution("Triangle Classify Coverage Test", test_cases, test_results, show_result=True)
    
    all_statements = set(range(1, 16))
    all_branches = {f"{i}{j}" for i in range(1, 9) for j in ["T", "F"]}
    
    print_coverage_summary(total_covered_stmts, total_covered_branches, all_statements, all_branches)
    print_branch_coverage_table(test_cases, test_results, max_branch_num=8)
    print()
    
    print_latex_table(
        test_cases,
        test_results,
        max_branch_num=8,
        caption="Branch coverage for Triangle.classify() by test case ($\\times$ = covered)"
    )
    
    return 0


if __name__ == "__main__":
    exit(main())