Corrected output of compute mutations --eem

evolbioinfo · Jan 15, 2024 · b1a0a71 · b1a0a71
1 parent 72dcae6
commit b1a0a71
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/cmd/extractmutations.go b/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 {

diff --git a/test.sh b/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
-Original file line number
+Diff line change
@@ Expand Up / @@ -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	G	T	2
 	0	C	A	2
 	0	G	C	1
@@ Expand Down @@