def test_QobjEigenStates(): "Qobj eigenstates" data = np.eye(5) A = Qobj(data) b, c = A.eigenstates() assert_(np.all(b - np.ones(5) == 0)) kets = [basis(5, k) for k in range(5)] for k in range(5): assert_(c[k] == kets[k])