Add unit test with pytest

1 files changed, 1 insertions, 1 deletions

diff --git a/solvers.py b/solvers.py
index b544f1b..040c6f4 100644
--- a/solvers.py
+++ b/solvers.py
@@ -114,4 +114,4 @@ def gaussian_eliminate(
     y = forward_substitution(ll, pp @ bb)
     # U @ x = y
     x = back_substitution(uu, y)
-    return x
+    return x[:, 0]