From 1b3e6f1d8dad0b3b7d8a0891dcb2050cdc023075 Mon Sep 17 00:00:00 2001
From: Eric Wieser <wieser.eric@gmail.com>
Date: Sat, 1 Jul 2017 11:40:29 +0100
Subject: BUG: np.resize discards empty shapes

---
 numpy/core/fromnumeric.py | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

(limited to 'numpy/core/fromnumeric.py')

diff --git a/numpy/core/fromnumeric.py b/numpy/core/fromnumeric.py
index a8c2fd2fb..4922fc3e4 100644
--- a/numpy/core/fromnumeric.py
+++ b/numpy/core/fromnumeric.py
@@ -1120,18 +1120,16 @@ def resize(a, new_shape):
         new_shape = (new_shape,)
     a = ravel(a)
     Na = len(a)
-    if not Na:
-        return mu.zeros(new_shape, a.dtype)
     total_size = um.multiply.reduce(new_shape)
+    if Na == 0 or total_size == 0:
+        return mu.zeros(new_shape, a.dtype)
+
     n_copies = int(total_size / Na)
     extra = total_size % Na
 
-    if total_size == 0:
-        return a[:0]
-
     if extra != 0:
-        n_copies = n_copies+1
-        extra = Na-extra
+        n_copies = n_copies + 1
+        extra = Na - extra
 
     a = concatenate((a,)*n_copies)
     if extra > 0:
-- 
cgit v1.2.1