From 9e64ee042c07933ed8e818a4589c8f08314500dd Mon Sep 17 00:00:00 2001 From: jbrockmendel Date: Wed, 11 Mar 2020 16:58:31 -0700 Subject: [PATCH] ENH: implement EA.size --- pandas/core/arrays/base.py | 7 +++++++ pandas/tests/extension/base/interface.py | 3 +++ 2 files changed, 10 insertions(+) diff --git a/pandas/core/arrays/base.py b/pandas/core/arrays/base.py index b5da6d4c11616..6aa303dd04703 100644 --- a/pandas/core/arrays/base.py +++ b/pandas/core/arrays/base.py @@ -407,6 +407,13 @@ def shape(self) -> Tuple[int, ...]: """ return (len(self),) + @property + def size(self) -> int: + """ + The number of elements in the array. + """ + return np.prod(self.shape) + @property def ndim(self) -> int: """ diff --git a/pandas/tests/extension/base/interface.py b/pandas/tests/extension/base/interface.py index cdea96334be2a..95fb3d7439b56 100644 --- a/pandas/tests/extension/base/interface.py +++ b/pandas/tests/extension/base/interface.py @@ -19,6 +19,9 @@ class BaseInterfaceTests(BaseExtensionTests): def test_len(self, data): assert len(data) == 100 + def test_size(self, data): + assert data.size == 100 + def test_ndim(self, data): assert data.ndim == 1