Allow custom names for lhs/rhs

You often want to give the columns more meaningful names than lhs/rhs
when you merge multiple datasets.

git-svn-id: https://llvm.org/svn/llvm-project/test-suite/trunk@313582 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 6 insertions, 1 deletions

diff --git a/utils/compare.py b/utils/compare.py
index 90710d01..b6578770 100755
--- a/utils/compare.py
+++ b/utils/compare.py
@@ -249,6 +249,10 @@ if __name__ == "__main__":
                         dest='merge_function', const=pd.DataFrame.min)
     parser.add_argument('--merge-max', action='store_const',
                         dest='merge_function', const=pd.DataFrame.max)
+    parser.add_argument('--lhs-name', default="lhs",
+                        help="Name used to describe left side in 'vs' mode")
+    parser.add_argument('--rhs-name', default="rhs",
+                        help="Name used to describe right side in 'vs' mode")
     parser.add_argument('files', metavar='FILE', nargs='+')
     config = parser.parse_args()
 
@@ -269,7 +273,8 @@ if __name__ == "__main__":
         rhs_merged = config.merge_function(rhs_d, level=1)
 
         # Combine to new dataframe
-        data = pd.concat([lhs_merged, rhs_merged], names=['l/r'], keys=['lhs', 'rhs'])
+        data = pd.concat([lhs_merged, rhs_merged], names=['l/r'],
+                         keys=[config.lhs_name, config.rhs_name])
     else:
         data = readmulti(files)