Python 2 and 3 behave differently. On python 3 i get an error when doing: `plot_schmidt(tensor(basis(2, 1), basis(2, 0)))`