Make module simpler

cov/__init__.py

@@ -1 +1 @@
-from .utils import *
+from .cov import *

cov/__main__.py

@@ -1,13 +1,13 @@
 import sys
 import pytest
-import cov.utils
+import cov.cov
 
 
 def main() -> None:
     arguments = sys.argv
 
     pytest.main(arguments[1:])
-    cov.utils.print_results()
+    cov.cov.print_table()
 
 
 if __name__ == "__main__":

cov/cov.py

@@ -0,0 +1,46 @@
+from typing import List, Dict, Callable
+from tabulate import tabulate
+import inspect
+
+__all__ = ["test", "mark"]
+
+
+class Coverage:
+    def __init__(self, n_branches: int) -> None:
+        self.n_branches: int = n_branches
+        self.covered_branches: List[bool] = n_branches * [False]
+
+    def get_coverage(self) -> int:
+        return self.covered_branches.count(True) * 100 // self.n_branches
+
+    def get_missing(self) -> str:
+        missing = [str(branch_id) for branch_id, covered in enumerate(self.covered_branches) if covered is False]
+        return " ".join(missing)
+
+
+coverages: Dict[str, Coverage] = dict()
+
+
+def test(n_branches: int) -> Callable:
+    def decorator(func: Callable) -> Callable:
+        coverages[func.__name__] = Coverage(n_branches)
+        return func
+
+    return decorator
+
+
+def mark(branch_id: int) -> None:
+    func_name = inspect.stack()[1].function
+    coverage = coverages[func_name]
+
+    coverage.covered_branches[branch_id] = True
+
+
+def print_table() -> None:
+    header = ["Function", "Branch %", "Branches", "Missing"]
+    rows = [
+        (func_name, coverage.get_coverage(), coverage.n_branches, coverage.get_missing())
+        for func_name, coverage in coverages.items()
+    ]
+
+    print(tabulate(rows, header))