update unittests

Azure · May 14, 2024 · 9b21837 · 9b21837
1 parent d673269
commit 9b21837
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 9 deletions.
diff --git a/packages/autorest.python/test/unittests/test_m2r.py b/packages/autorest.python/test/unittests/test_m2r.py
@@ -3,7 +3,7 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from autorest.m2r import M2R
+from pygen.m2r import M2R
 
 
 _MD_LINK = "[inline link](https://github.com/Azure/autorest.python)"

diff --git a/packages/autorest.python/test/unittests/test_name_converter.py b/packages/autorest.python/test/unittests/test_name_converter.py
@@ -3,8 +3,8 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from autorest.preprocess import PreProcessPlugin
-from autorest.preprocess.python_mappings import PadType
+from pygen.preprocess import PreProcessPlugin
+from pygen.preprocess.python_mappings import PadType
 
 
 def pad_reserved_words(name: str, pad_type: PadType) -> str:

diff --git a/packages/autorest.python/test/unittests/test_optional_return_type.py b/packages/autorest.python/test/unittests/test_optional_return_type.py
@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------
 
 import pytest
-from autorest.codegen.models import (
+from pygen.codegen.models import (
     Operation,
     LROOperation,
     PagingOperation,
@@ -15,8 +15,8 @@
     RequestBuilder,
     Client,
 )
-from autorest.codegen.models.parameter_list import RequestBuilderParameterList
-from autorest.codegen.models.primitive_types import StringType
+from pygen.codegen.models.parameter_list import RequestBuilderParameterList
+from pygen.codegen.models.primitive_types import StringType
 
 
 @pytest.fixture

diff --git a/packages/autorest.python/test/unittests/test_parameter_ordering.py b/packages/autorest.python/test/unittests/test_parameter_ordering.py
@@ -3,8 +3,8 @@
 # Licensed under the MIT License. See License.txt in the project root for
 # license information.
 # --------------------------------------------------------------------------
-from autorest.codegen.models import Parameter, AnyType, CodeModel, StringType
-from autorest.codegen.models.parameter_list import ParameterList
+from pygen.codegen.models import Parameter, AnyType, CodeModel, StringType
+from pygen.codegen.models.parameter_list import ParameterList
 
 
 def get_code_model():

diff --git a/packages/autorest.python/test/unittests/test_sort_schema.py b/packages/autorest.python/test/unittests/test_sort_schema.py
@@ -4,7 +4,7 @@
 # license information.
 # --------------------------------------------------------------------------
 
-from autorest.codegen.models import CodeModel, DPGModelType
+from pygen.codegen.models import CodeModel, DPGModelType
 
 
 def get_code_model():