def test_state_inner1(): state = ar(10, 10) state = State(state) inner = state.inner() assert np.allclose(inner, state.inner(state)) inner = state.inner(diagonal=False) assert np.allclose(inner, state.inner(state, diagonal=False))
def test_state_inner(): state = ar(10, 10) state = State(state) inner = state.inner(diag=False) assert np.allclose(inner, state.inner(state, diag=False)) inner_diag = state.inner() assert np.allclose(np.diag(inner), inner_diag)
def test_state_inner_differing_size(): state1 = State(ar(8, 10)) state2 = State(ar(4, 10)) inner = state1.inner(state2, diagonal=False) assert inner.shape == (8, 4) inner = state1.inner(state2) assert inner.shape == (4, )
def test_state_inner_matrix(): state = ar(10, 10) M = ar(10) state = State(state) inner = state.inner(matrix=M, diag=False) assert np.allclose(inner, state.inner(state, matrix=np.diag(M), diag=False)) inner_diag = state.inner(matrix=M) assert np.allclose(np.diag(inner), inner_diag) inner_diag = state.inner(matrix=np.diag(M)) assert np.allclose(np.diag(inner), inner_diag)