Serial flag python

schema/check_generated_data.py

@@ -12,6 +12,10 @@
 import os.path
 import sys
 
+# To get commanlin arguments
+sys.path.append('../testdriver')
+from ddtargs import schemaArgs
+
 import schema_validator
 from schema_files import ALL_TEST_TYPES
 
@@ -101,6 +105,5 @@ def main(args):
         logging.info("All %d generated test data files match with schema", schema_count)
 
 
-
 if __name__ == "__main__":
     main(sys.argv)

schema/check_schemas.py

@@ -15,6 +15,10 @@
 import schema_validator
 from schema_files import ALL_TEST_TYPES
 
+# To get commanlin arguments
+sys.path.append('../testdriver')
+from ddtargs import schemaArgs
+
 
 class ValidateSchema:
     def __init__(self, schema_base='.'):
@@ -61,17 +65,26 @@ def save_schema_validation_summary(self, validation_status):
 
 
 def parallel_validate_schema(validator, file_names):
-    num_processors = multiprocessing.cpu_count()
-    logging.info('Schema validation: %s processors for %s schema validations', num_processors, len(file_names))
-
-    processor_pool = multiprocessing.Pool(num_processors)
-    # How to get all the results
-    result = None
-    try:
-        result = processor_pool.map(validator.validate_schema_file, file_names)
-    except multiprocessing.pool.MaybeEncodingError as error:
-        pass
-    return result
+    if not validator.options.run_serial:
+        num_processors = multiprocessing.cpu_count()
+        logging.info('Schema validation: %s processors for %s schema validations',
+                     num_processors, len(file_names))
+
+        processor_pool = multiprocessing.Pool(num_processors)
+        # How to get all the results
+        result = None
+        try:
+            result = processor_pool.map(validator.validate_schema_file, file_names)
+        except multiprocessing.pool.MaybeEncodingError as error:
+            pass
+        return result
+    else:
+        results = []
+        logging.info('check_schemas running serially on %s files!',
+                     len(filenames))
+        for file_name in file_names:
+            results.append(validator.validate_schema_file(file_name))
+        return results
 
 
 def main(args):
@@ -83,6 +96,8 @@ def main(args):
     # Todo: use setters to initialize validator
     validator.schema_base = '.'
 
+    validator.options = SchemaArgs(args).getOptions()
+
     if len(args) > 1:
         schema_base = args[1]
     else:

schema/check_test_output.py

@@ -14,6 +14,10 @@
 import schema_files
 from schema_files import ALL_TEST_TYPES
 
+# To get commanlin arguments
+sys.path.append('../testdriver')
+from ddtargs import schemaArgs
+
 
 def main(args):
     logging.config.fileConfig("../logging.conf")

testdriver/ddtargs.py

@@ -103,6 +103,21 @@ def getOptions(self):
     return self.options
 
 
+class schemaArgs():
+  def __init__(self, args):
+    self.parser = argparse.ArgumentParser(
+        description='Schema check arguments')
+
+    self.parser.add_argument('--run_serial', default=None,
+                             help='Set if execution should be done serially. Parallel is the default.')
+    self.options = self.parser.parse_args(args)
+
+
+
+  def getOptions(self):
+    return self.options
+
+
 # Set up arguments common to both testDriver and verifier
 def setCommonArgs(parser):