From 795d25d9a7b21712321a306fd748d103eee7b0f8 Mon Sep 17 00:00:00 2001
From: David Halter <davidhalter88@gmail.com>
Date: Sun, 6 Jan 2013 01:48:48 +0100
Subject: [PATCH] improved some run.py stuff again

---
 test/base.py | 14 ++++++++++++++
 test/run.py  | 15 ++-------------
 2 files changed, 16 insertions(+), 13 deletions(-)

diff --git a/test/base.py b/test/base.py
index 82f0a382..91a19d11 100644
--- a/test/base.py
+++ b/test/base.py
@@ -32,6 +32,20 @@ sys.argv = sys.argv[:1] + args
 summary = []
 tests_fail = 0
 
+def get_test_list():
+# get test list, that should be executed
+    test_files = {}
+    last = None
+    for arg in sys.argv[1:]:
+        if arg.isdigit():
+            if last is None:
+                continue
+            test_files[last].append(int(arg))
+        else:
+            test_files[arg] = []
+            last = arg
+    return test_files
+
 class TestBase(unittest.TestCase):
     def get_script(self, src, pos, path=None):
         if pos is None:
diff --git a/test/run.py b/test/run.py
index fc7d6e37..3e1d7361 100755
--- a/test/run.py
+++ b/test/run.py
@@ -154,9 +154,8 @@ def run_test(source, f_name, lines_to_execute):
     correct = None
     test_type = None
     start = None
-    for line_nr, line in enumerate(StringIO(source)):
+    for line_nr, line in enumerate(StringIO(source), 1):
         line = unicode(line)
-        line_nr += 1
         if correct:
             r = re.match('^(\d+)\s*(.*)$', correct)
             if r:
@@ -240,17 +239,7 @@ try:
 except ValueError:
     thirdparty = False
 
-# get test list, that should be executed
-test_files = {}
-last = None
-for arg in sys.argv[1:]:
-    if arg.isdigit():
-        if last is None:
-            continue
-        test_files[last].append(int(arg))
-    else:
-        test_files[arg] = []
-        last = arg
+test_files = base.get_test_list()
 
 # completion tests:
 completion_test_dir = '../test/completion'