CSIT-768: Refactor Python container libraries

[csit.git] / resources / tools / doc_gen / gen_rst.py

diff --git a/resources/tools/doc_gen/gen_rst.py b/resources/tools/doc_gen/gen_rst.py
index e3356f5..898f215 100755 (executable)
--- a/resources/tools/doc_gen/gen_rst.py
+++ b/resources/tools/doc_gen/gen_rst.py
@@ -163,7 +163,7 @@ def write_module_title(fh, module_name):
     :type fh: file
     :type module_name: str
     """
-    title = "{} module".format(module_name)
+    title = "{} suite".format(module_name)
     fh.write("\n{}\n{}\n".format(title, '-' * len(title)))
 
 
@@ -190,15 +190,16 @@ def generate_py_rst_files():
         with open(full_path, mode='a') as fh:
             if getsize(full_path) == 0:
                 package = file_name.split('.')[-2]
-                fh.write("{} package\n".format(package))
-                fh.write('=' * len("{} package".format(package)))
+                fh.write("{}\n".format(package))
+                fh.write('=' * len("{}".format(package)))
             module_path = file_name.split('.')[:-1]
             if dirs:
                 write_toc(fh, module_path, dirs)
             for file in files:
                 module_name = file.split('.')[0]
                 write_module_title(fh, module_name)
-                fh.write(rst_py_module.format('.'.join(module_path), module_name))
+                fh.write(rst_py_module.format('.'.join(module_path),
+                                              module_name))
 
 
 def generate_rf_rst_files(file_names, incl_tests=True, incl_keywords=True):
@@ -222,8 +223,8 @@ def generate_rf_rst_files(file_names, incl_tests=True, incl_keywords=True):
         with open(full_path, mode='a') as fh:
             if getsize(full_path) == 0:
                 package = file_name.split('.')[-2]
-                fh.write("{} package\n".format(package))
-                fh.write('=' * len("{} package".format(package)) + '\n')
+                fh.write("{}\n".format(package))
+                fh.write('=' * len("{}".format(package)) + '\n')
             module_path = file_name.split('.')[:-1]
             if dirs:
                 write_toc(fh, module_path, dirs)