diff --git a/pymongo/__init__.py b/pymongo/__init__.py index 9a35750811..95eabef242 100644 --- a/pymongo/__init__.py +++ b/pymongo/__init__.py @@ -106,7 +106,14 @@ from pymongo.write_concern import WriteConcern # Public module compatibility imports -import pymongo.uri_parser # noqa: F401 # isort: skip +# isort: off +from pymongo import uri_parser # noqa: F401 +from pymongo import change_stream # noqa: F401 +from pymongo import client_session # noqa: F401 +from pymongo import collection # noqa: F401 +from pymongo import command_cursor # noqa: F401 +from pymongo import database # noqa: F401 +# isort: on version = __version__ """Current version of PyMongo.""" diff --git a/test/test_default_exports.py b/test/test_default_exports.py index 5f3e749d36..adc3882a36 100644 --- a/test/test_default_exports.py +++ b/test/test_default_exports.py @@ -215,6 +215,12 @@ def test_pymongo_submodule_attributes(self): self.assertTrue(hasattr(pymongo, "uri_parser")) self.assertTrue(pymongo.uri_parser) self.assertTrue(pymongo.uri_parser.parse_uri) + self.assertTrue(pymongo.change_stream) + self.assertTrue(pymongo.client_session) + self.assertTrue(pymongo.collection) + self.assertTrue(pymongo.cursor) + self.assertTrue(pymongo.command_cursor) + self.assertTrue(pymongo.database) def test_gridfs_imports(self): import gridfs