From b1a0a7146c8de242a3471fd8ccb47e5e0b9c3608 Mon Sep 17 00:00:00 2001
From: Frederic Lemoine <fredericlemoine@users.noreply.github.com>
Date: Mon, 15 Jan 2024 23:57:44 +0100
Subject: [PATCH] Corrected output of compute mutations --eem

---
 cmd/extractmutations.go | 7 +++++--
 test.sh                 | 2 +-
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/cmd/extractmutations.go b/cmd/extractmutations.go
index 278e628..61876bd 100644
--- 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
index 7c872cf..807d21c 100755
--- 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