From 0e12b4825a4a1827a062b932b65fb4460f9a4288 Mon Sep 17 00:00:00 2001 From: Andres Guzman-Ballen Date: Fri, 17 Jul 2026 12:25:12 -0500 Subject: [PATCH] Add test asserting cuda.core. is not imported --- cuda_core/tests/test_duplicate_imports.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 cuda_core/tests/test_duplicate_imports.py diff --git a/cuda_core/tests/test_duplicate_imports.py b/cuda_core/tests/test_duplicate_imports.py new file mode 100644 index 0000000000..101877a094 --- /dev/null +++ b/cuda_core/tests/test_duplicate_imports.py @@ -0,0 +1,20 @@ +# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# +# SPDX-License-Identifier: Apache-2.0 + +"""Tests cuda.core.__init__.py does not import duplicate modules.""" + +import sys + +from cuda import bindings + +cuda_major = bindings.__version__.split(".")[0] + + +def test_typing_module_imports(): + """ + Importing cuda.core.system should not also import cuda.core.cuXX.system + """ + + assert "cuda.core.system" in sys.modules + assert f"cuda.core.cu{cuda_major}.system" not in sys.modules