Changed 1-d slicing operation - now drops dimension always

Author: dwest77a

arraypartition/partition.py

@@ -119,6 +119,10 @@ def shape(self):
         if not self._extent:
             return self._shape
         for d, e in enumerate(self._extent):
+
+            if isinstance(e, int):
+                continue
+            
             start = e.start or 0
             stop  = e.stop or self._shape[d]
             step  = e.step or 1

arraypartition/tests/test_array.py

@@ -22,7 +22,7 @@ def test_array(self):
 
         assert ap.shape == (20,20), "Shape Error"
 
-        assert (np.array(array_part)[5][0] - 0.463) < 0.001, "Data Error"
+        assert (np.array(array_part)[0][5][0] - 0.463) < 0.001, "Data Error"
         assert (np.array(ap)[5][0] - 0.803) < 0.001, "Data Error"
 
 if __name__ == '__main__':