Add blank lines in higher-dim array formatting

Author: jturner314

src/arrayformat.rs

@@ -102,6 +102,7 @@ where
 
             let n_to_be_printed = to_be_printed.len();
 
+            let blank_lines = "\n".repeat(shape.len() - 2);
             let indent = " ".repeat(depth + 1);
 
             write!(f, "[")?;
@@ -123,10 +124,10 @@ where
                         // We need to add a separator after each slice,
                         // apart from the last one
                         if j != n_to_be_printed - 1 {
-                            write!(f, ",\n")?
+                            write!(f, ",\n{}", blank_lines)?
                         }
                     }
-                    PrintableCell::Ellipses => write!(f, "{}...,\n", indent)?,
+                    PrintableCell::Ellipses => write!(f, "{}...,\n{}", indent, blank_lines)?,
                 }
             }
             write!(f, "]")?;

tests/format.rs

@@ -15,7 +15,7 @@ fn formatting() {
     let a = a.reshape((4, 1, 1));
     assert_eq!(
         format!("{:4}", a),
-        "[[[   1]],\n [[   2]],\n [[   3]],\n [[   4]]]"
+        "[[[   1]],\n\n [[   2]],\n\n [[   3]],\n\n [[   4]]]",
     );
 
     let a = a.reshape((2, 2));