Fix DisplayFmt

1 files changed, 9 insertions, 4 deletions

diff --git a/libpsi/src/maths/matrix.rs b/libpsi/src/maths/matrix.rs
index 795b37f..975c92d 100644
--- a/libpsi/src/maths/matrix.rs
+++ b/libpsi/src/maths/matrix.rs
@@ -1,5 +1,3 @@
-// TODO(Hachem): Refactor
-
 use super::Float;
 use core::{fmt, ops};
 
@@ -175,10 +173,17 @@ impl<T: Float + fmt::Debug> fmt::Debug for Matrix<T> {
 impl<T: Float + fmt::Display> fmt::Display for Matrix<T> {
     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
         for i in 0..self.rows {
+            write!(f, "[")?;
             for j in 0..self.cols {
-                write!(f, "{:>8} ", self.get(i, j))?;
+                write!(f, "{}", self.get(i, j))?;
+                if j != self.cols - 1 {
+                    write!(f, " ")?;
+                }
+            }
+            write!(f, "]")?;
+            if i != self.rows - 1 {
+                write!(f, "\n")?;
             }
-            writeln!(f)?;
         }
         Ok(())
     }