def test_initialization():
# Test Kalman filter initialization
mod = Representation(1, k_states=2)
# Test invalid state initialization
with pytest.raises(RuntimeError):
mod._initialize_state()
# Test valid initialization
initial_state = np.zeros(2,) + 1.5
initial_state_cov = np.eye(2) * 3.
mod.initialize_known(initial_state, initial_state_cov)
assert_equal(mod.initialization.constant.sum(), 3)
assert_equal(mod.initialization.stationary_cov.diagonal().sum(), 6)
# Test invalid initial_state
initial_state = np.zeros(10,)
with pytest.raises(ValueError):
mod.initialize_known(initial_state, initial_state_cov)
initial_state = np.zeros((10, 10))
with pytest.raises(ValueError):
mod.initialize_known(initial_state, initial_state_cov)
# Test invalid initial_state_cov
initial_state = np.zeros(2,) + 1.5
initial_state_cov = np.eye(3)
with pytest.raises(ValueError):
mod.initialize_known(initial_state, initial_state_cov)
def test_initialization():
# Test Kalman filter initialization
mod = Representation(1, k_states=2)
# Test invalid state initialization
assert_raises(RuntimeError, lambda: mod._initialize_state())
# Test valid initialization
initial_state = np.zeros(2) + 1.5
initial_state_cov = np.eye(2) * 3.0
mod.initialize_known(initial_state, initial_state_cov)
assert_equal(mod._initial_state.sum(), 3)
assert_equal(mod._initial_state_cov.diagonal().sum(), 6)
# Test invalid initial_state
initial_state = np.zeros(10)
assert_raises(ValueError, lambda: mod.initialize_known(initial_state, initial_state_cov))
initial_state = np.zeros((10, 10))
assert_raises(ValueError, lambda: mod.initialize_known(initial_state, initial_state_cov))
# Test invalid initial_state_cov
initial_state = np.zeros(2) + 1.5
initial_state_cov = np.eye(3)
assert_raises(ValueError, lambda: mod.initialize_known(initial_state, initial_state_cov))
def test_initialization():
# Test Kalman filter initialization
mod = Representation(1, k_states=2)
# Test invalid state initialization
assert_raises(RuntimeError, lambda: mod._initialize_state())
# Test valid initialization
initial_state = np.zeros(2, ) + 1.5
initial_state_cov = np.eye(2) * 3.
mod.initialize_known(initial_state, initial_state_cov)
assert_equal(mod._initial_state.sum(), 3)
assert_equal(mod._initial_state_cov.diagonal().sum(), 6)
# Test invalid initial_state
initial_state = np.zeros(10, )
assert_raises(
ValueError,
lambda: mod.initialize_known(initial_state, initial_state_cov))
initial_state = np.zeros((10, 10))
assert_raises(
ValueError,
lambda: mod.initialize_known(initial_state, initial_state_cov))
# Test invalid initial_state_cov
initial_state = np.zeros(2, ) + 1.5
initial_state_cov = np.eye(3)
assert_raises(
ValueError,
lambda: mod.initialize_known(initial_state, initial_state_cov))
