def _mks_fcoeff(x_data, n_space, n_state):
return fit(
x_data,
_response(x_data, n_space, n_state),
discretize(n_state=n_state),
redundancy_func=redundancy,
)
def test():
"""Very simple example."""
assert np.allclose(
fit(
_get_x(),
_get_x().swapaxes(1, 2),
discretize(n_state=2),
redundancy_func=redundancy,
),
[[[0.5, 0.5], [-2, 0]], [[-0.5, 0], [-1, 0]]],
)