Corrected output of compute mutations --eem

Author: fredericlemoine

cmd/extractmutations.go

@@ -93,8 +93,11 @@ var mutationsCmd = &cobra.Command{
 		}
 		defer closeWriteFile(f, outfile)
 
-		fmt.Fprintf(f, "Tree ID\tSite\tBranch ID\tNode Name\tParent Character\tChild Character\tTotal tips\tSame Character Tips\n")
-
+		if mutationseems {
+			fmt.Fprintf(f, "Tree ID\tSite\tParent Character\tChild Character\tNb EEMs\n")
+		} else {
+			fmt.Fprintf(f, "Tree ID\tSite\tBranch ID\tNode Name\tParent Character\tChild Character\tTotal tips\tSame Character Tips\n")
+		}
 		for t := range treechan {
 			if mutationseems {
 				if muts, err = mutations.CountEEMs(t.Tree, align); err != nil {

test.sh

@@ -678,7 +678,7 @@ cat > input_tree <<EOF
 EOF
 
 cat > expected <<EOF
-Tree ID	Site	Branch ID	Node Name	Parent Character	Child Character	Total tips	Same Character Tips
+Tree ID	Site	Parent Character	Child Character	Nb EEMs
 0	0	G	T	2
 0	0	C	A	2
 0	0	G	C	1