Просмотр исходного кода

Replace map operations with list comprehensions.

nuwan_ag 10 лет назад
Родитель
Сommit
d551762f96
1 измененных файлов с 10 добавлено и 6 удалено
  1. 10 6
      test/helpers.py

+ 10 - 6
test/helpers.py

@@ -182,10 +182,12 @@ class ProviderTestCaseGenerator():
         class
         class
         """
         """
         suite = unittest.TestSuite()
         suite = unittest.TestSuite()
-        suites = map(
-            lambda test_class: self.generate_test_suite_for_provider_testcase(
-                provider_class, test_class), self.all_test_classes)
-        map(suite.addTest, suites)
+        suites = [
+            self.generate_test_suite_for_provider_testcase(
+                provider_class, test_class)
+            for test_class in self.all_test_classes]
+        for s in suites:
+            suite.addTest(s)
         return suite
         return suite
 
 
     def generate_tests(self):
     def generate_tests(self):
@@ -210,6 +212,8 @@ class ProviderTestCaseGenerator():
             provider_classes = factory.get_all_provider_classes(
             provider_classes = factory.get_all_provider_classes(
                 get_mock=use_mock_drivers)
                 get_mock=use_mock_drivers)
         suite = unittest.TestSuite()
         suite = unittest.TestSuite()
-        suites = map(self.generate_test_suite_for_provider, provider_classes)
-        map(suite.addTest, suites)
+        suites = [
+            self.generate_test_suite_for_provider(p) for p in provider_classes]
+        for s in suites:
+            suite.addTest(s)
         return suite
         return suite