mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-20 03:44:11 +08:00
1421551297
* Cython KF1D to Python * cleanup * set x * less nesting * fix release * Revert "fix release" This reverts commit 97e5d0f80405660efb861c7b9df7fbfd5dda5dd9.
36 lines
732 B
Python
36 lines
732 B
Python
import unittest
|
|
|
|
from openpilot.common.simple_kalman import KF1D
|
|
|
|
|
|
class TestSimpleKalman(unittest.TestCase):
|
|
def setUp(self):
|
|
dt = 0.01
|
|
x0_0 = 0.0
|
|
x1_0 = 0.0
|
|
A0_0 = 1.0
|
|
A0_1 = dt
|
|
A1_0 = 0.0
|
|
A1_1 = 1.0
|
|
C0_0 = 1.0
|
|
C0_1 = 0.0
|
|
K0_0 = 0.12287673
|
|
K1_0 = 0.29666309
|
|
|
|
self.kf = KF1D(x0=[[x0_0], [x1_0]],
|
|
A=[[A0_0, A0_1], [A1_0, A1_1]],
|
|
C=[C0_0, C0_1],
|
|
K=[[K0_0], [K1_0]])
|
|
|
|
def test_getter_setter(self):
|
|
self.kf.set_x([[1.0], [1.0]])
|
|
self.assertEqual(self.kf.x, [[1.0], [1.0]])
|
|
|
|
def update_returns_state(self):
|
|
x = self.kf.update(100)
|
|
self.assertEqual(x, self.kf.x)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|