From f1ca811abb26b9ade70cb80ddc2dc8ed89b330b3 Mon Sep 17 00:00:00 2001 From: Georg Brandl Date: Sun, 25 Oct 2009 20:25:43 +0000 Subject: [PATCH] Fix a demo. --- Demo/comparisons/sortingtest.py | 14 ++++---------- 1 file changed, 4 insertions(+), 10 deletions(-) diff --git a/Demo/comparisons/sortingtest.py b/Demo/comparisons/sortingtest.py index cabf6260d9c..08a73e3754d 100755 --- a/Demo/comparisons/sortingtest.py +++ b/Demo/comparisons/sortingtest.py @@ -24,7 +24,6 @@ # - Handles blank input lines correctly import re -import string import sys def main(): @@ -32,18 +31,13 @@ def main(): def makekey(item, prog=prog): match = prog.match(item) if match: - var, num = match.group(1, 2) - return string.atoi(num), var + var, num = match.groups() + return int(num), var else: # Bad input -- pretend it's a var with value 0 return 0, item - while 1: - line = sys.stdin.readline() - if not line: - break - items = line.split() - items = map(makekey, items) - items.sort() + for line in sys.stdin: + items = sorted(makekey(item) for item in line.split()) for num, var in items: print "%s=%s" % (var, num), print