Update test_linear.py

mathLab · Nov 15, 2023 · 1b6ac68 · 1b6ac68
1 parent 58cd332
commit 1b6ac68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_linear.py b/tests/test_linear.py
@@ -56,7 +56,7 @@ def test_with_db_predict(self):
         db = Database(np.array([1, 2, 3]), Y)
         rom = ReducedOrderModel(db, POD(), Linear())
         rom.fit()
-        assert rom.predict([1.]).shape == 3
+        assert rom.predict([1.]).shape == (3,)
 
 
     def test_wrong1(self):