MAINT: Address attribute error in trim_zeros

Author: lagru

numpy/lib/function_base.py

@@ -1651,7 +1651,7 @@ def trim_zeros(filt, trim='fb', axis=-1):
         if 'b' not in trim:
             stop = (None,) * ndim
 
-    if start.size == 1:
+    if len(start) == 1:
         # filt is 1D -> don't use multi-dimensional slicing to preserve
         # non-array input types
         sl = slice(start[0], stop[0])

numpy/lib/tests/test_function_base.py

@@ -1146,6 +1146,18 @@ def test_allzero(self, ndim):
         res = trim_zeros(a, axis=None)
         assert_array_equal(res, np.zeros((0,) * ndim))
 
+    def test_trim_arg(self):
+        a = np.array([0, 1, 2, 0])
+
+        res = trim_zeros(a, trim='f')
+        assert_array_equal(res, [1, 2, 0])
+
+        res = trim_zeros(a, trim='b')
+        assert_array_equal(res, [0, 1, 2])
+
+        res = trim_zeros(a, trim='')
+        assert_array_equal(res, [0, 1, 2, 0])
+
 
 class TestExtins(object):