diff --git a/schulze.py b/schulze.py index 59099a4..f567ae7 100644 --- a/schulze.py +++ b/schulze.py @@ -80,7 +80,7 @@ def _rank_p(candidate_names, p): candidate_wins[num_wins].append(candidate_name1) - sorted_wins = sorted(candidate_wins.iterkeys(), reverse=True) + sorted_wins = sorted(candidate_wins.keys(), reverse=True) return [candidate_wins[num_wins] for num_wins in sorted_wins] diff --git a/schulze_test.py b/schulze_test.py index 5fc6e69..bc105aa 100644 --- a/schulze_test.py +++ b/schulze_test.py @@ -17,7 +17,7 @@ def _make_ranks(self, names): def _assert_row(self, matrix, name, expected_values): other_names = (other_name for other_name in 'abcde' if other_name != name) - for other_name, expected_value in itertools.izip(other_names, expected_values): + for other_name, expected_value in zip(other_names, expected_values): actual_value = matrix.get((name, other_name), 0) self.assertEqual(expected_value, actual_value, 'matrix(%s, %s)=%s, expected %s' %