#---------------- |0+> = |00>+|01>
system_dims = (2,2)
init = np.zeros(np.prod(system_dims), dtype=complex)
init[0] = 1
init[1] = 1
test_srv(system_dims, init, [1, 1])
#----------------Bell, |00>+|11>
system_dims = (2,2)
init = np.zeros(np.prod(system_dims), dtype=complex)
init[0] = 1
init[3] = 1
test_srv(system_dims, init, [2, 2])
#----------------GHZ, |000>+|111>
system_dims = (2,2,2)
init = np.zeros(np.prod(system_dims), dtype=complex)
init[0] = 1
init[7] = 1
test_srv(system_dims, init, [2,2,2])
#----------------Sym, |000>+|111>+|222>
system_dims = (3,3,3)
init = np.zeros(np.prod(system_dims), dtype=complex)
init[0] = 1
init[13] = 1
init[26] = 1
test_srv(system_dims, init, [3,3,3])
#----------------Wikipedia example, |000>+|101>+|210>+|311>
system_dims = (4,4,4)
init = np.zeros(np.prod(system_dims), dtype=complex)
init[0] = 1
init[17] = 1
init[36] = 1
init[53] = 1
test_srv(system_dims, init, [2, 2, 4])
passed test, svr: [2, 2, 2]
passed test, svr: [3, 3, 3]
$$\[\begin{align}
|\psi\rangle =
\begin{bmatrix}
\frac{\sqrt{3}}{3} & 0 & 0 & 0 & \cdots & 0 & 0 & \frac{\sqrt{3}}{3} \\
\end{bmatrix}
\\
\text{dims=(3, 3, 3)}
\end{align}\]$$
passed test, svr: [2, 2, 4]
$$\[\begin{align}
|\psi\rangle =
\begin{bmatrix}
\frac{1}{2} & 0 & 0 & 0 & \cdots & 0 & 0 & 0 \\
\end{bmatrix}
\\
\text{dims=(4, 4, 4)}
\end{align}\]$$