Skip to content

Commit 3798ee3

Browse files
committed
added unit test
1 parent 9141871 commit 3798ee3

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

tests/unit/test_imports.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
import codegen
2+
from codegen import Codebase
3+
4+
5+
def test_codegen_imports():
6+
# Test decorated function
7+
@codegen.function(name="sample_codemod")
8+
def run(codebase):
9+
pass
10+
11+
# Test class
12+
cls = codegen.Function
13+
assert cls is not None
14+
15+
codebase = Codebase("./")
16+
assert codebase is not None

0 commit comments

Comments
 (0)