refactor: let TimeSeriesParam return array

smallbunnies · Jan 12, 2025 · 79c132f · 79c132f
1 parent 9fe0317
commit 79c132f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Solverz/equation/param.py b/Solverz/equation/param.py
@@ -140,9 +140,9 @@ def get_v_t(self, t):
             # input of interp1d is zero-dimensional, we need to reshape
             # [0] is to eliminate the numpy DeprecationWarning in m3b9 test: Conversion of an array with ndim > 0 to a scalar is
             # deprecated, and will error in the future, which should be resolved.
-            vt = self.vt(t).reshape((-1,))[0]
+            vt = self.vt(t).reshape((-1,))
         else:
-            vt = self.v_series[-1]
+            vt = self.v_series[-1:]
 
         if self.index is not None:
             temp = self.v.copy()

diff --git a/Solverz/equation/test/test_Param.py b/Solverz/equation/test/test_Param.py
@@ -46,7 +46,7 @@ def test_Param():
                          v_series=[0, 10, 100],
                          time_series=[0, 10, 20],
                          value=0)
-    assert Pb.get_v_t(5).__str__() == '5.0'
+    assert Pb.get_v_t(5).__str__() == '[5.]'
 
     try:
         Pb = TimeSeriesParam(name='Pb',