Skip to content

Commit f5799ac

Browse files
committed
Add type checking to movetest.py
1 parent c45231d commit f5799ac

2 files changed

Lines changed: 115 additions & 81 deletions

File tree

rope/contrib/generate.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from typing import Literal, Optional
1919

2020
from rope.base.project import Project
21-
from rope.base.resources import Resource
21+
from rope.base.resources import Resource, File, Folder
2222

2323
GenerateKind = Literal[
2424
"variable",
@@ -51,7 +51,7 @@ def create_generate(
5151
return generate(project, resource, offset, goal_resource=goal_resource)
5252

5353

54-
def create_module(project, name, sourcefolder=None):
54+
def create_module(project, name, sourcefolder=None) -> File:
5555
"""Creates a module and returns a `rope.base.resources.File`"""
5656
if sourcefolder is None:
5757
sourcefolder = project.root
@@ -62,7 +62,7 @@ def create_module(project, name, sourcefolder=None):
6262
return parent.create_file(packages[-1] + ".py")
6363

6464

65-
def create_package(project, name, sourcefolder=None):
65+
def create_package(project, name, sourcefolder=None) -> Folder:
6666
"""Creates a package and returns a `rope.base.resources.Folder`"""
6767
if sourcefolder is None:
6868
sourcefolder = project.root

0 commit comments

Comments
 (0)