Skip to content

Commit

Permalink
Merge pull request #19501 from charris/backport-19456
Browse files Browse the repository at this point in the history
TST: Fix a `GenericAlias` test failure for python 3.9.0
  • Loading branch information
charris committed Jul 17, 2021
2 parents 74f82c9 + c87de48 commit 4c80d18
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion numpy/typing/tests/test_generic_alias.py
Expand Up @@ -60,7 +60,6 @@ class TestGenericAlias:
("__call__", lambda n: n(shape=(1,), dtype=np.int64, buffer=BUFFER)),
("subclassing", lambda n: _get_subclass_mro(n)),
("pickle", lambda n: n == pickle.loads(pickle.dumps(n))),
("__weakref__", lambda n: n == weakref.ref(n)()),
])
def test_pass(self, name: str, func: FuncType) -> None:
"""Compare `types.GenericAlias` with its numpy-based backport.
Expand All @@ -75,6 +74,14 @@ def test_pass(self, name: str, func: FuncType) -> None:
value_ref = func(NDArray_ref)
assert value == value_ref

def test_weakref(self) -> None:
"""Test ``__weakref__``."""
value = weakref.ref(NDArray)()

if sys.version_info >= (3, 9, 1): # xref bpo-42332
value_ref = weakref.ref(NDArray_ref)()
assert value == value_ref

@pytest.mark.parametrize("name", GETATTR_NAMES)
def test_getattr(self, name: str) -> None:
"""Test that `getattr` wraps around the underlying type,
Expand Down

0 comments on commit 4c80d18

Please sign in to comment.