diff --git a/client/tests/mtproto_test.py b/client/tests/mtproto_test.py index 6b184ca8..8ec99cd4 100644 --- a/client/tests/mtproto_test.py +++ b/client/tests/mtproto_test.py @@ -2,6 +2,7 @@ import struct from typing import Optional from pytest import raises + from telethon._impl.crypto import AuthKey from telethon._impl.mtproto import Encrypted, Plain, RpcError from telethon._impl.mtproto.mtp.types import MsgId diff --git a/client/tests/mtsender_test.py b/client/tests/mtsender_test.py index 93ab7cee..0a9e2c59 100644 --- a/client/tests/mtsender_test.py +++ b/client/tests/mtsender_test.py @@ -2,6 +2,7 @@ import asyncio import logging from pytest import LogCaptureFixture, mark + from telethon._impl.mtproto import Full from telethon._impl.mtsender import connect from telethon._impl.tl import LAYER, abcs, functions, types diff --git a/client/tests/peer_ref_test.py b/client/tests/peer_ref_test.py index 6a23246c..9c22bd7e 100644 --- a/client/tests/peer_ref_test.py +++ b/client/tests/peer_ref_test.py @@ -1,6 +1,7 @@ import inspect from pytest import raises + from telethon._impl.session import ChannelRef, GroupRef, PeerRef, UserRef USER = UserRef(12, 34) diff --git a/client/tests/reader_test.py b/client/tests/reader_test.py index 26037bfc..ca8ac16c 100644 --- a/client/tests/reader_test.py +++ b/client/tests/reader_test.py @@ -1,6 +1,7 @@ import struct from pytest import mark + from telethon._impl.tl.core import Reader, Serializable from telethon._impl.tl.mtproto.types import BadServerSalt from telethon._impl.tl.types import GeoPoint diff --git a/client/tests/rsa_test.py b/client/tests/rsa_test.py index 90986925..fe791fe9 100644 --- a/client/tests/rsa_test.py +++ b/client/tests/rsa_test.py @@ -1,4 +1,5 @@ from rsa import PublicKey + from telethon._impl.crypto.rsa import ( PRODUCTION_RSA_KEY, TESTMODE_RSA_KEY, diff --git a/client/tests/serializable_test.py b/client/tests/serializable_test.py index f402186e..6ffbef41 100644 --- a/client/tests/serializable_test.py +++ b/client/tests/serializable_test.py @@ -1,4 +1,5 @@ from pytest import mark + from telethon._impl.tl.core import serialize_bytes_to diff --git a/client/tests/transport/abridged_test.py b/client/tests/transport/abridged_test.py index d3b425ab..0ef5ca60 100644 --- a/client/tests/transport/abridged_test.py +++ b/client/tests/transport/abridged_test.py @@ -1,4 +1,5 @@ from pytest import raises + from telethon._impl.mtproto import Abridged diff --git a/client/tests/transport/full_test.py b/client/tests/transport/full_test.py index 79441a86..c1e7f9bc 100644 --- a/client/tests/transport/full_test.py +++ b/client/tests/transport/full_test.py @@ -1,4 +1,5 @@ from pytest import raises + from telethon._impl.mtproto import Full diff --git a/client/tests/transport/intermediate_test.py b/client/tests/transport/intermediate_test.py index e9d55f39..52fe4761 100644 --- a/client/tests/transport/intermediate_test.py +++ b/client/tests/transport/intermediate_test.py @@ -1,4 +1,5 @@ from pytest import raises + from telethon._impl.mtproto import Intermediate diff --git a/client/tests/two_factor_auth_test.py b/client/tests/two_factor_auth_test.py index 40a60b82..32df3fba 100644 --- a/client/tests/two_factor_auth_test.py +++ b/client/tests/two_factor_auth_test.py @@ -1,4 +1,5 @@ from pytest import mark, raises + from telethon._impl.crypto.two_factor_auth import ( calculate_2fa, check_p_prime_and_subgroup, diff --git a/client/tests/types_test.py b/client/tests/types_test.py index e19dc957..34bebe52 100644 --- a/client/tests/types_test.py +++ b/client/tests/types_test.py @@ -1,4 +1,5 @@ from pytest import mark + from telethon._impl.client.types import AdminRight from telethon._impl.tl import types diff --git a/generator/tests/common_test.py b/generator/tests/common_test.py index f94be28d..616bf073 100644 --- a/generator/tests/common_test.py +++ b/generator/tests/common_test.py @@ -1,4 +1,5 @@ from pytest import mark + from telethon_generator._impl.codegen.serde.common import ( split_words, to_class_name, diff --git a/generator/tests/definition_test.py b/generator/tests/definition_test.py index e395d77e..1f3d3909 100644 --- a/generator/tests/definition_test.py +++ b/generator/tests/definition_test.py @@ -1,4 +1,5 @@ from pytest import mark, raises + from telethon_generator.tl_parser import ( Definition, Flag, diff --git a/generator/tests/parameter_test.py b/generator/tests/parameter_test.py index 61829f30..5d7d1b37 100644 --- a/generator/tests/parameter_test.py +++ b/generator/tests/parameter_test.py @@ -1,4 +1,5 @@ from pytest import mark, raises + from telethon_generator.tl_parser import ( Flag, FlagsParameter, diff --git a/generator/tests/tl_iterator_test.py b/generator/tests/tl_iterator_test.py index 2e2bb6e6..b4d41bc4 100644 --- a/generator/tests/tl_iterator_test.py +++ b/generator/tests/tl_iterator_test.py @@ -1,4 +1,5 @@ from pytest import raises + from telethon_generator.tl_parser import FunctionDef, TypeDef, parse_tl_file diff --git a/generator/tests/ty_test.py b/generator/tests/ty_test.py index a797c759..ac1f5879 100644 --- a/generator/tests/ty_test.py +++ b/generator/tests/ty_test.py @@ -1,6 +1,7 @@ from typing import Optional from pytest import mark, raises + from telethon_generator.tl_parser import Type