Merge pull request #53 from rohaquinlop/issue-52

feat(build): #52 improve usability of the package
rohaquinlop · Oct 28, 2024 · 65edd78 · 65edd78
2 parents 38f8871 + 173f41a
commit 65edd78
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/complexipy/__init__.py b/complexipy/__init__.py
@@ -0,0 +1,18 @@
+from complexipy.rust import (
+    CodeComplexity,
+    FileComplexity,
+    FunctionComplexity,
+)
+
+from .main import (
+    code_complexity,
+    file_complexity,
+)
+
+__all__ = [
+    "CodeComplexity",
+    "FileComplexity",
+    "FunctionComplexity",
+    "code_complexity",
+    "file_complexity",
+]
diff --git a/tests/main.py b/tests/main.py
@@ -2,7 +2,10 @@
 import unittest
 from pathlib import Path
 
-from complexipy.main import file_complexity, code_complexity
+from complexipy import (
+    file_complexity,
+    code_complexity,
+)
 
 
 class TestFiles(unittest.TestCase):