Update imports in tensor tests and add conftest.py

Author: vlad-perevezentsev

dpctl_ext/tests/tensor/conftest.py

@@ -0,0 +1,31 @@
+# *****************************************************************************
+# Copyright (c) 2026, Intel Corporation
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+# - Redistributions of source code must retain the above copyright notice,
+#   this list of conditions and the following disclaimer.
+# - Redistributions in binary form must reproduce the above copyright notice,
+#   this list of conditions and the following disclaimer in the documentation
+#   and/or other materials provided with the distribution.
+# - Neither the name of the copyright holder nor the names of its contributors
+#   may be used to endorse or promote products derived from this software
+#   without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
+# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
+# THE POSSIBILITY OF SUCH DAMAGE.
+# *****************************************************************************
+
+"""Configures pytest to discover helper/ module"""
+
+from dpnp.tests.conftest import suppress_invalid_numpy_warnings

dpctl_ext/tests/tensor/elementwise/test_abs.py

@@ -35,16 +35,17 @@
 # TODO: revert to `import dpctl.tensor...`
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
-from dpnp.tests.tensor.elementwise.utils import (
+
+from ..helper import (
+    get_queue_or_skip,
+    skip_if_dtype_not_supported,
+)
+from .utils import (
     _all_dtypes,
     _complex_fp_dtypes,
     _real_fp_dtypes,
     _usm_types,
 )
-from dpnp.tests.tensor.helper import (
-    get_queue_or_skip,
-    skip_if_dtype_not_supported,
-)
 
 
 @pytest.mark.parametrize("dtype", _all_dtypes)

dpctl_ext/tests/tensor/elementwise/test_add.py

@@ -38,15 +38,16 @@
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
 from dpctl_ext.tensor._type_utils import _can_cast
-from dpnp.tests.tensor.elementwise.utils import (
+
+from ..helper import (
+    get_queue_or_skip,
+    skip_if_dtype_not_supported,
+)
+from .utils import (
     _all_dtypes,
     _compare_dtypes,
     _usm_types,
 )
-from dpnp.tests.tensor.helper import (
-    get_queue_or_skip,
-    skip_if_dtype_not_supported,
-)
 
 
 @pytest.mark.parametrize("op1_dtype", _all_dtypes)

dpctl_ext/tests/tensor/elementwise/test_angle.py

@@ -35,15 +35,16 @@
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
 from dpctl_ext.tensor._type_utils import _can_cast
-from dpnp.tests.tensor.elementwise.utils import (
+
+from ..helper import (
+    get_queue_or_skip,
+    skip_if_dtype_not_supported,
+)
+from .utils import (
     _all_dtypes,
     _complex_fp_dtypes,
     _no_complex_dtypes,
 )
-from dpnp.tests.tensor.helper import (
-    get_queue_or_skip,
-    skip_if_dtype_not_supported,
-)
 
 
 @pytest.mark.parametrize("dtype", _all_dtypes)

dpctl_ext/tests/tensor/elementwise/test_atan2.py

@@ -35,14 +35,15 @@
 # TODO: revert to `import dpctl.tensor...`
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
-from dpnp.tests.tensor.elementwise.utils import (
-    _compare_dtypes,
-    _no_complex_dtypes,
-)
-from dpnp.tests.tensor.helper import (
+
+from ..helper import (
     get_queue_or_skip,
     skip_if_dtype_not_supported,
 )
+from .utils import (
+    _compare_dtypes,
+    _no_complex_dtypes,
+)
 
 
 @pytest.mark.parametrize("op1_dtype", _no_complex_dtypes[1:])

dpctl_ext/tests/tensor/elementwise/test_bitwise_and.py

@@ -33,11 +33,12 @@
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
 from dpctl_ext.tensor._type_utils import _can_cast
-from dpnp.tests.tensor.elementwise.utils import _integral_dtypes
-from dpnp.tests.tensor.helper import (
+
+from ..helper import (
     get_queue_or_skip,
     skip_if_dtype_not_supported,
 )
+from .utils import _integral_dtypes
 
 
 @pytest.mark.parametrize("op_dtype", _integral_dtypes)

dpctl_ext/tests/tensor/elementwise/test_bitwise_invert.py

@@ -32,15 +32,16 @@
 # TODO: revert to `import dpctl.tensor...`
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
-from dpnp.tests.tensor.elementwise.utils import (
+
+from ..helper import (
+    get_queue_or_skip,
+    skip_if_dtype_not_supported,
+)
+from .utils import (
     _compare_dtypes,
     _integral_dtypes,
     _usm_types,
 )
-from dpnp.tests.tensor.helper import (
-    get_queue_or_skip,
-    skip_if_dtype_not_supported,
-)
 
 
 @pytest.mark.parametrize(

dpctl_ext/tests/tensor/elementwise/test_bitwise_left_shift.py

@@ -33,11 +33,12 @@
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
 from dpctl_ext.tensor._type_utils import _can_cast
-from dpnp.tests.tensor.elementwise.utils import _integral_dtypes
-from dpnp.tests.tensor.helper import (
+
+from ..helper import (
     get_queue_or_skip,
     skip_if_dtype_not_supported,
 )
+from .utils import _integral_dtypes
 
 
 @pytest.mark.parametrize("op1_dtype", _integral_dtypes)

dpctl_ext/tests/tensor/elementwise/test_bitwise_or.py

@@ -33,11 +33,12 @@
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
 from dpctl_ext.tensor._type_utils import _can_cast
-from dpnp.tests.tensor.elementwise.utils import _integral_dtypes
-from dpnp.tests.tensor.helper import (
+
+from ..helper import (
     get_queue_or_skip,
     skip_if_dtype_not_supported,
 )
+from .utils import _integral_dtypes
 
 
 @pytest.mark.parametrize("op_dtype", _integral_dtypes)

dpctl_ext/tests/tensor/elementwise/test_bitwise_right_shift.py

@@ -33,11 +33,12 @@
 # when dpnp fully migrates dpctl/tensor
 import dpctl_ext.tensor as dpt
 from dpctl_ext.tensor._type_utils import _can_cast
-from dpnp.tests.tensor.elementwise.utils import _integral_dtypes
-from dpnp.tests.tensor.helper import (
+
+from ..helper import (
     get_queue_or_skip,
     skip_if_dtype_not_supported,
 )
+from .utils import _integral_dtypes
 
 
 @pytest.mark.parametrize("op1_dtype", _integral_dtypes)