File tree Expand file tree Collapse file tree 2 files changed +11
-1
lines changed Expand file tree Collapse file tree 2 files changed +11
-1
lines changed Original file line number Diff line number Diff line change @@ -85,6 +85,9 @@ def _create_package(self) -> None:
8585 package_init_template = self .env .get_template ("package_init.pyi" )
8686 package_init .write_text (package_init_template .render (description = self .package_description ))
8787
88+ pytyped = self .package_dir / "py.typed"
89+ pytyped .write_text ("# Marker file for PEP 561" )
90+
8891 def _build_metadata (self ) -> None :
8992 # Create a pyproject.toml file
9093 pyproject_template = self .env .get_template ("pyproject.toml" )
Original file line number Diff line number Diff line change @@ -184,14 +184,21 @@ def test__create_package(self, mocker):
184184 project .env = mocker .MagicMock ()
185185 project .env .get_template .return_value = package_init_template
186186 package_init_path = mocker .MagicMock (autospec = pathlib .Path )
187- project .package_dir .__truediv__ .return_value = package_init_path
187+ pytyped_path = mocker .MagicMock (autospec = pathlib .Path )
188+ paths = {
189+ "__init__.py" : package_init_path ,
190+ "py.typed" : pytyped_path ,
191+ }
192+
193+ project .package_dir .__truediv__ .side_effect = lambda x : paths [x ]
188194
189195 project ._create_package ()
190196
191197 project .package_dir .mkdir .assert_called_once ()
192198 project .env .get_template .assert_called_once_with ("package_init.pyi" )
193199 package_init_template .render .assert_called_once_with (description = project .package_description )
194200 package_init_path .write_text .assert_called_once_with (package_init_template .render ())
201+ pytyped_path .write_text .assert_called_once_with ("# Marker file for PEP 561" )
195202
196203 def test__build_metadata (self , mocker ):
197204 from openapi_python_client import _Project
You can’t perform that action at this time.
0 commit comments