[PATCH] D51328: Fix lit/example/many-tests pickling issue

[Nathan Lanza via Phabricator via llvm-commits]

lanza updated this revision to Diff 165427.
lanza added a comment.

format


Repository:
  rL LLVM

https://reviews.llvm.org/D51328

Files:
  utils/lit/examples/many-tests/ManyTests.py
  utils/lit/examples/many-tests/lit.cfg


Index: utils/lit/examples/many-tests/lit.cfg
===================================================================
--- utils/lit/examples/many-tests/lit.cfg
+++ utils/lit/examples/many-tests/lit.cfg
@@ -1,23 +1,8 @@
 # -*- Python -*-
 
-from lit import Test
+import sys
+import os
+sys.path.insert(0, os.path.dirname(__file__))
+import ManyTests
 
-class ManyTests(object):
-    def __init__(self, N=10000):
-        self.N = N
-
-    def getTestsInDirectory(self, testSuite, path_in_suite,
-                            litConfig, localConfig):
-        for i in range(self.N):
-            test_name = 'test-%04d' % (i,)
-            yield Test.Test(testSuite, path_in_suite + (test_name,),
-                            localConfig)
-
-    def execute(self, test, litConfig):
-        # Do a "non-trivial" amount of Python work.
-        sum = 0
-        for i in range(10000):
-            sum += i
-        return Test.PASS,''
-
-config.test_format = ManyTests()
+config.test_format = ManyTests.ManyTests()
Index: utils/lit/examples/many-tests/ManyTests.py
===================================================================
--- utils/lit/examples/many-tests/ManyTests.py
+++ utils/lit/examples/many-tests/ManyTests.py
@@ -1,23 +1,18 @@
-# -*- Python -*-
-
 from lit import Test
 
+
 class ManyTests(object):
     def __init__(self, N=10000):
         self.N = N
 
-    def getTestsInDirectory(self, testSuite, path_in_suite,
-                            litConfig, localConfig):
+    def getTestsInDirectory(self, testSuite, path_in_suite, litConfig, localConfig):
         for i in range(self.N):
-            test_name = 'test-%04d' % (i,)
-            yield Test.Test(testSuite, path_in_suite + (test_name,),
-                            localConfig)
+            test_name = "test-%04d" % (i,)
+            yield Test.Test(testSuite, path_in_suite + (test_name,), localConfig)
 
     def execute(self, test, litConfig):
         # Do a "non-trivial" amount of Python work.
         sum = 0
         for i in range(10000):
             sum += i
-        return Test.PASS,''
-
-config.test_format = ManyTests()
+        return Test.PASS, ""


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D51328.165427.patch
Type: text/x-patch
Size: 2145 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20180914/2476359b/attachment.bin>