def test_update_n_basis_modes_unfit_basis(data_random):
data = data_random
n_basis_modes = 5
model = SSPOR()
model.update_n_basis_modes(n_basis_modes, data)
assert model.basis_matrix_.shape[1] == n_basis_modes
def test_update_n_basis_modes_refit(data_random):
data = data_random
n_basis_modes = 5
model = SSPOR(basis=Identity(n_basis_modes=n_basis_modes))
model.fit(data)
assert model.basis_matrix_.shape[1] == n_basis_modes
model.update_n_basis_modes(n_basis_modes + 1, data)
assert model.basis_matrix_.shape[1] == n_basis_modes + 1
def test_update_n_basis_modes(data_random):
data = data_random
model = SSPOR()
model.fit(data)
assert model.basis.n_basis_modes == data.shape[0]
assert model.basis_matrix_.shape[1] == data.shape[0]
n_basis_modes = 5
model.update_n_basis_modes(n_basis_modes)
assert model.basis.n_basis_modes == data.shape[0]
assert model.basis_matrix_.shape[1] == n_basis_modes
def test_update_n_basis_modes_errors(data_random):
data = data_random
n_basis_modes = 5
model = SSPOR(basis=Identity(n_basis_modes=n_basis_modes))
model.fit(data)
with pytest.raises(ValueError):
model.update_n_basis_modes(0)
with pytest.raises(ValueError):
model.update_n_basis_modes("5")
with pytest.raises(ValueError):
model.update_n_basis_modes(data.shape[0] + 1)
# Need to pass x when increasing n_basis_modes beyond capacity
# of the original basis
with pytest.raises(ValueError):
model.update_n_basis_modes(n_basis_modes + 1)