From 2ce2eb627d14dbff42bb932c852f0407d82d13d2 Mon Sep 17 00:00:00 2001 From: Sebastien Date: Fri, 26 Jul 2024 17:29:35 +1000 Subject: [PATCH] fix: removing future warning in sp3 tests. --- gnssanalysis/gn_io/sp3.py | 8 ++++++-- tests/test_sp3.py | 2 +- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/gnssanalysis/gn_io/sp3.py b/gnssanalysis/gn_io/sp3.py index 608b172..897d7aa 100644 --- a/gnssanalysis/gn_io/sp3.py +++ b/gnssanalysis/gn_io/sp3.py @@ -275,7 +275,9 @@ def getVelSpline(sp3Df: _pd.DataFrame) -> _pd.DataFrame: sp3dfECI = sp3Df.EST.unstack(1)[["X", "Y", "Z"]] # _ecef2eci(sp3df) datetime = sp3dfECI.index.get_level_values("J2000").values spline = _interpolate.CubicSpline(datetime, sp3dfECI.values) - velDf = _pd.DataFrame(data=spline.derivative(1)(datetime), index=sp3dfECI.index, columns=sp3dfECI.columns).stack(1) + velDf = _pd.DataFrame(data=spline.derivative(1)(datetime), index=sp3dfECI.index, columns=sp3dfECI.columns).stack( + 1, future_stack=True + ) return _pd.concat([sp3Df, _pd.concat([velDf], keys=["VELi"], axis=1)], axis=1) @@ -309,7 +311,9 @@ def getVelPoly(sp3Df: _pd.DataFrame, deg: int = 35) -> _pd.DataFrame: res_prev = coeff.T.dot(inputs_prev) res_next = coeff.T.dot(inputs_next) - vel_i = _pd.DataFrame((((y - res_prev.T) + (res_next.T - y)) / 2), columns=est.columns, index=est.index).stack() + vel_i = _pd.DataFrame((((y - res_prev.T) + (res_next.T - y)) / 2), columns=est.columns, index=est.index).stack( + future_stack=True + ) vel_i.columns = [["VELi"] * 3] + [vel_i.columns.values.tolist()] diff --git a/tests/test_sp3.py b/tests/test_sp3.py index bc131ec..08518a2 100644 --- a/tests/test_sp3.py +++ b/tests/test_sp3.py @@ -86,6 +86,6 @@ def test_velinterpolation(self, mock_file): """ result = sp3.read_sp3("mock_path", pOnly=True) r = sp3.getVelSpline(result) - r2 = sp3.getVelPoly(result, 3) + r2 = sp3.getVelPoly(result, 2) self.assertIsNotNone(r) self.assertIsNotNone(r2)