| 1234567891011121314151617181920212223242526272829303132333435363738394041424344 |
- import pytest
- import warnings
- import numpy as np
- @pytest.mark.filterwarnings("error")
- def test_getattr_warning():
- # issue gh-14735: make sure we clear only getattr errors, and let warnings
- # through
- class Wrapper:
- def __init__(self, array):
- self.array = array
- def __len__(self):
- return len(self.array)
- def __getitem__(self, item):
- return type(self)(self.array[item])
- def __getattr__(self, name):
- if name.startswith("__array_"):
- warnings.warn("object got converted", UserWarning, stacklevel=1)
- return getattr(self.array, name)
- def __repr__(self):
- return "<Wrapper({self.array})>".format(self=self)
- array = Wrapper(np.arange(10))
- with pytest.raises(UserWarning, match="object got converted"):
- np.asarray(array)
- def test_array_called():
- class Wrapper:
- val = '0' * 100
- def __array__(self, result=None):
- return np.array([self.val], dtype=object)
- wrapped = Wrapper()
- arr = np.array(wrapped, dtype=str)
- assert arr.dtype == 'U100'
- assert arr[0] == Wrapper.val
|
