def test_eigs(): backend = shell_backend.ShellBackend() init = shell_backend.ShellTensor((2, ), np.float64) eta, v = backend.eigs(lambda x: x, initial_state=init, numeig=2) assert len(eta) == 2 for n in range(len(eta)): assert v[n].shape == (2, ) def mv(x): return x eta, v = backend.eigs(mv, shape=(2, ), dtype=np.float64, numeig=2) assert len(eta) == 2 for n in range(len(eta)): assert v[n].shape == (2, )
def test_shell_tensor_reshape(): shell_tensor = shell_backend.ShellTensor((2, 1), np.float64) shell_tensor = shell_tensor.reshape((1, 2)) assert shell_tensor.shape == (1, 2)