Skip to content

Commit ad5f714

Browse files
committed
Add __array__ method to mock numpy-like
1 parent 0ecb8c2 commit ad5f714

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

pandas/tests/dtypes/test_inference.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -79,14 +79,16 @@ def __iter__(self):
7979
iter_values = iter(self._values)
8080

8181
def it_outer():
82-
for element in iter_values:
83-
yield element
82+
yield from iter_values
8483

8584
return it_outer()
8685

8786
def __len__(self):
8887
return len(self._values)
8988

89+
def __array__(self, t=None):
90+
return self._values
91+
9092
@property
9193
def ndim(self):
9294
return self._values.ndim
@@ -210,10 +212,13 @@ class DtypeList(list):
210212
assert not inference.is_array_like(123)
211213

212214

213-
@pytest.mark.parametrize("eg", (
214-
np.array(2),
215-
MockNumpyLikeArray(np.array(2)),
216-
))
215+
@pytest.mark.parametrize(
216+
"eg",
217+
(
218+
np.array(2),
219+
MockNumpyLikeArray(np.array(2)),
220+
),
221+
)
217222
def test_assert_almost_equal(eg):
218223
tm.assert_almost_equal(eg, eg)
219224

0 commit comments

Comments
 (0)