diff --git a/kwant/tests/test_system.py b/kwant/tests/test_system.py index 346a2dd4691165c8e08c85e8fd76cf96843010aa..6fa5cd2cee7b9f00df4de8917256d5b392371362 100644 --- a/kwant/tests/test_system.py +++ b/kwant/tests/test_system.py @@ -91,10 +91,10 @@ def test_hamiltonian_submatrix(): sys[((gr(i), gr(i + 1)) for i in xrange(2))] = hopping sys2 = sys.finalized() mat = sys2.hamiltonian_submatrix((2, 1)) - mat_should_be = [[5, 1, 0], [1, 4, 1.], [0, 1, 3]] + mat_should_be = [[3, 1, 0], [1, 4, 1], [0, 1, 5]] # Sorting is required due to unknown compression order of builder. - onsite_hamiltonians = mat.flat[::3] + onsite_hamiltonians = mat.flat[::4] perm = np.argsort(onsite_hamiltonians) mat = mat[perm, :] mat = mat[:, perm]