Skip to content

Commit 75d82dd

Browse files
Update tensor tests: use dpnp.tensor import
1 parent 1bca6b2 commit 75d82dd

86 files changed

Lines changed: 201 additions & 321 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

dpnp/tests/tensor/elementwise/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,5 +28,5 @@
2828

2929
"""
3030
Collection of test and utility files for testing elementwise operations
31-
over :class:`dpctl.tensor.usm_ndarray`.
31+
over :class:`dpnp.tensor.usm_ndarray`.
3232
"""

dpnp/tests/tensor/elementwise/test_abs.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,12 +29,11 @@
2929
import itertools
3030
import warnings
3131

32-
# TODO: revert to `import dpctl.tensor...`
33-
# when dpnp fully migrates dpctl/tensor
34-
import dpctl_ext.tensor as dpt
3532
import numpy as np
3633
import pytest
3734

35+
import dpnp.tensor as dpt
36+
3837
from ..helper import (
3938
get_queue_or_skip,
4039
skip_if_dtype_not_supported,

dpnp/tests/tensor/elementwise/test_add.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,12 @@
3030
import re
3131

3232
import dpctl
33-
34-
# TODO: revert to `import dpctl.tensor...`
35-
# when dpnp fully migrates dpctl/tensor
36-
import dpctl_ext.tensor as dpt
3733
import numpy as np
3834
import pytest
3935
from dpctl.utils import ExecutionPlacementError
40-
from dpctl_ext.tensor._type_utils import _can_cast
36+
37+
import dpnp.tensor as dpt
38+
from dpnp.tensor._type_utils import _can_cast
4139

4240
from ..helper import (
4341
get_queue_or_skip,

dpnp/tests/tensor/elementwise/test_angle.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,12 +28,11 @@
2828

2929
import itertools
3030

31-
# TODO: revert to `import dpctl.tensor...`
32-
# when dpnp fully migrates dpctl/tensor
33-
import dpctl_ext.tensor as dpt
3431
import numpy as np
3532
import pytest
36-
from dpctl_ext.tensor._type_utils import _can_cast
33+
34+
import dpnp.tensor as dpt
35+
from dpnp.tensor._type_utils import _can_cast
3736

3837
from ..helper import (
3938
get_queue_or_skip,

dpnp/tests/tensor/elementwise/test_atan2.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,12 @@
2828

2929
import ctypes
3030

31-
# TODO: revert to `import dpctl.tensor...`
32-
# when dpnp fully migrates dpctl/tensor
33-
import dpctl_ext.tensor as dpt
3431
import numpy as np
3532
import pytest
3633
from numpy.testing import assert_allclose
3734

35+
import dpnp.tensor as dpt
36+
3837
from ..helper import (
3938
get_queue_or_skip,
4039
skip_if_dtype_not_supported,

dpnp/tests/tensor/elementwise/test_bitwise_and.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,11 @@
2626
# THE POSSIBILITY OF SUCH DAMAGE.
2727
# *****************************************************************************
2828

29-
# TODO: revert to `import dpctl.tensor...`
30-
# when dpnp fully migrates dpctl/tensor
31-
import dpctl_ext.tensor as dpt
3229
import numpy as np
3330
import pytest
34-
from dpctl_ext.tensor._type_utils import _can_cast
31+
32+
import dpnp.tensor as dpt
33+
from dpnp.tensor._type_utils import _can_cast
3534

3635
from ..helper import (
3736
get_queue_or_skip,

dpnp/tests/tensor/elementwise/test_bitwise_invert.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,11 @@
2626
# THE POSSIBILITY OF SUCH DAMAGE.
2727
# *****************************************************************************
2828

29-
# TODO: revert to `import dpctl.tensor...`
30-
# when dpnp fully migrates dpctl/tensor
31-
import dpctl_ext.tensor as dpt
3229
import numpy as np
3330
import pytest
3431

32+
import dpnp.tensor as dpt
33+
3534
from ..helper import (
3635
get_queue_or_skip,
3736
skip_if_dtype_not_supported,

dpnp/tests/tensor/elementwise/test_bitwise_left_shift.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,11 @@
2626
# THE POSSIBILITY OF SUCH DAMAGE.
2727
# *****************************************************************************
2828

29-
# TODO: revert to `import dpctl.tensor...`
30-
# when dpnp fully migrates dpctl/tensor
31-
import dpctl_ext.tensor as dpt
3229
import numpy as np
3330
import pytest
34-
from dpctl_ext.tensor._type_utils import _can_cast
31+
32+
import dpnp.tensor as dpt
33+
from dpnp.tensor._type_utils import _can_cast
3534

3635
from ..helper import (
3736
get_queue_or_skip,

dpnp/tests/tensor/elementwise/test_bitwise_or.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,11 @@
2626
# THE POSSIBILITY OF SUCH DAMAGE.
2727
# *****************************************************************************
2828

29-
# TODO: revert to `import dpctl.tensor...`
30-
# when dpnp fully migrates dpctl/tensor
31-
import dpctl_ext.tensor as dpt
3229
import numpy as np
3330
import pytest
34-
from dpctl_ext.tensor._type_utils import _can_cast
31+
32+
import dpnp.tensor as dpt
33+
from dpnp.tensor._type_utils import _can_cast
3534

3635
from ..helper import (
3736
get_queue_or_skip,

dpnp/tests/tensor/elementwise/test_bitwise_right_shift.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,12 +26,11 @@
2626
# THE POSSIBILITY OF SUCH DAMAGE.
2727
# *****************************************************************************
2828

29-
# TODO: revert to `import dpctl.tensor...`
30-
# when dpnp fully migrates dpctl/tensor
31-
import dpctl_ext.tensor as dpt
3229
import numpy as np
3330
import pytest
34-
from dpctl_ext.tensor._type_utils import _can_cast
31+
32+
import dpnp.tensor as dpt
33+
from dpnp.tensor._type_utils import _can_cast
3534

3635
from ..helper import (
3736
get_queue_or_skip,

0 commit comments

Comments
 (0)