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)
