diff options
| author | Charles Harris <charlesr.harris@gmail.com> | 2021-02-27 17:36:29 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-02-27 17:36:29 -0700 |
| commit | cefab37bfba72ae062a58be3c7bdbd59c2c4d309 (patch) | |
| tree | 7c8b0e1e3b10fcdf9113be7c74171293bac039c8 /numpy | |
| parent | f9dd21e64584f6e821805daa450b8bc9f8ef3526 (diff) | |
| parent | 2195375e6b0ba74f772fecfdc7ca41b5897faed6 (diff) | |
| download | numpy-cefab37bfba72ae062a58be3c7bdbd59c2c4d309.tar.gz | |
Merge pull request #18497 from BvB93/einsumfunc
TST: Remove the `einsum` typing tests reliance on issuing a `ComplexWarning`
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/typing/tests/data/pass/einsumfunc.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/numpy/typing/tests/data/pass/einsumfunc.py b/numpy/typing/tests/data/pass/einsumfunc.py index 914eed4cc..a2a39fb1c 100644 --- a/numpy/typing/tests/data/pass/einsumfunc.py +++ b/numpy/typing/tests/data/pass/einsumfunc.py @@ -2,7 +2,6 @@ from __future__ import annotations from typing import List, Any -import pytest import numpy as np AR_LIKE_b = [True, True, True] @@ -12,6 +11,7 @@ AR_LIKE_f = [1.0, 2.0, 3.0] AR_LIKE_c = [1j, 2j, 3j] AR_LIKE_U = ["1", "2", "3"] +OUT_f: np.ndarray[Any, np.dtype[np.float64]] = np.empty(3, dtype=np.float64) OUT_c: np.ndarray[Any, np.dtype[np.complex128]] = np.empty(3, dtype=np.complex128) np.einsum("i,i->i", AR_LIKE_b, AR_LIKE_b) @@ -25,8 +25,7 @@ np.einsum("i,i,i,i->i", AR_LIKE_b, AR_LIKE_u, AR_LIKE_i, AR_LIKE_c) np.einsum("i,i->i", AR_LIKE_f, AR_LIKE_f, dtype="c16") np.einsum("i,i->i", AR_LIKE_U, AR_LIKE_U, dtype=bool, casting="unsafe") np.einsum("i,i->i", AR_LIKE_f, AR_LIKE_f, out=OUT_c) -with pytest.raises(np.ComplexWarning): - np.einsum("i,i->i", AR_LIKE_U, AR_LIKE_U, dtype=float, casting="unsafe", out=OUT_c) +np.einsum("i,i->i", AR_LIKE_U, AR_LIKE_U, dtype=int, casting="unsafe", out=OUT_f) np.einsum_path("i,i->i", AR_LIKE_b, AR_LIKE_b) np.einsum_path("i,i->i", AR_LIKE_u, AR_LIKE_u) |