Merge pull request #862 from jturner314/fix-zip-indexed-0dim

Fix Zip::indexed for the 0-dimensional case

Author: bluss

src/zip/mod.rs

@@ -691,12 +691,14 @@ impl<P, D> Zip<P, D>
 where
     D: Dimension,
 {
-    fn apply_core<F, Acc>(&mut self, acc: Acc, function: F) -> FoldWhile<Acc>
+    fn apply_core<F, Acc>(&mut self, acc: Acc, mut function: F) -> FoldWhile<Acc>
     where
         F: FnMut(Acc, P::Item) -> FoldWhile<Acc>,
         P: ZippableTuple<Dim = D>,
     {
-        if self.layout.is(CORDER | FORDER) {
+        if self.dimension.ndim() == 0 {
+            function(acc, unsafe { self.parts.as_ref(self.parts.as_ptr()) })
+        } else if self.layout.is(CORDER | FORDER) {
             self.apply_core_contiguous(acc, function)
         } else {
             self.apply_core_strided(acc, function)

tests/azip.rs

@@ -303,6 +303,19 @@ fn test_clone() {
     });
 }
 
+#[test]
+fn test_indices_0() {
+    let a1 = arr0(3);
+
+    let mut count = 0;
+    Zip::indexed(&a1).apply(|i, elt| {
+        count += 1;
+        assert_eq!(i, ());
+        assert_eq!(*elt, 3);
+    });
+    assert_eq!(count, 1);
+}
+
 #[test]
 fn test_indices_1() {
     let mut a1 = Array::default(12);